昇腾社区首页
中文
注册
开发者
下载

REG_AUTO_MAPPING_OP

产品支持情况

产品

是否支持

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品/Atlas A2 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

头文件/库文件

  • 头文件:#include <graph/custom.h>
  • 库文件:

功能说明

开发人员可以选择将自定义算子注册到框架中,由框架在编译最开始调用REG_AUTO_MAPPING_OP进行自定义算子注册。

函数原型

1
REG_AUTO_MAPPING_OP(custom_op_class)

参数说明

参数名

输入/输出

描述

custom_op_class

输入

自定义算子名称。

返回值说明

约束说明