跳到主要内容

REGISTER_CUSTOM_OP宏

宏功能

按指定名称注册算子。

宏原型

REGISTER_CUSTOM_OP(name)

参数说明

参数输入/输出说明
name输入算子类型名称。