REG_AUTO_MAPPING_OP
产品支持情况
产品 |
是否支持 |
|---|---|
√ |
|
√ |
|
√ |
|
√ |
|
√ |
头文件/库文件
- 头文件:#include <graph/custom.h>
- 库文件:
功能说明
开发人员可以选择将自定义算子注册到框架中,由框架在编译最开始调用REG_AUTO_MAPPING_OP进行自定义算子注册。
函数原型
1 | REG_AUTO_MAPPING_OP(custom_op_class) |
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
custom_op_class |
输入 |
自定义算子名称。 |
返回值说明
无
约束说明
无
父主题: 图基础数据结构和接口