更新时间:2021-10-26 GMT+08:00
Register函数
函数功能
将算子的类型保存到一个自定义的集合中。
函数原型
void Register(const std::string& op_type);
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
op_type |
输入 |
需要保存的算子类型 |
父主题: OpTypeContainer类