更新时间:2021-03-18 GMT+08:00
OpTypeRegistrar函数
函数功能
构造函数,将一个算子的类型注册到Container中。
函数原型
OpTypeRegistrar(const std::string& op_type);
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
op_type |
输入 |
需要注册的算子类型 |
父主题: OpTypeRegistrar类