更新时间:2021-10-26 GMT+08:00
分享

算子类型注册宏

宏功能

对一种算子类型进行注册,主要是声明一个常量字符串,该字符串就是算子类型的名称,然后将其保存到一个set中。

宏原型

REGISTER_OPTYPE_DECLARE(var_name, str_name)

REGISTER_OPTYPE_DEFINE(var_name, str_name)

参数说明

参数

输入/输出

说明

var_name

输入

常量字符串名称

str_name

输入

算子类型名称

分享:

    相关文档

    相关产品