文档首页 > > TBE自定义算子开发指南> 接口参考> 插件和原型定义接口> 算子原型接口> 算子原型注册宏>

INPUT

INPUT

分享
更新时间:2020/07/23 GMT+08:00

函数功能

注册算子输入信息。

注册算子输入信息成功后,自动生成算子输入的相关接口,用于获取算子输入的名称、设置算子输入的对应描述等。

例如,注册算子输入x,算子输入接收的数据类型为TensorType{DT_FLOAT},可调用INPUT(x, TensorType{DT_FLOAT})接口,注册算子输入成功后,自动生成以下相关接口:

static const string name_in_x(); // 返回输入的名称,即“x”
_THIS_TYPE& set_input_x(Operator& v, const string& srcName);// 指定输入x与算子对象v的输出srcName存在连接关系,返回算子对象本身
_THIS_TYPE& set_input_x(Operator& v); // 指定输入x与算子对象v的索引0的输出存在连接关系,返回算子对象本身
TensorDesc get_input_desc_x(); // 返回输入x对应的描述
graphStatus update_input_desc_x(const TensorDesc& tensorDesc);// 设置输入x对应的描述,包括Shape、DataType、Format等信息,graphStatus即uint32_t类型,返回非0表示出错

函数原型

INPUT (x, t)

参数说明

参数名

输入/输出

描述

x

输入

宏参数,算子输入的名称

t

输入

算子输入接收的数据类型,可以是TensorType定义的一个或多个,如果多个,通过“,”隔离,例如:

TensorType{DT_FLOAT}

TensorType({DT_FLOAT, DT_INT8}

关于TensorType类,请参见TensorType类说明

返回值

无。

异常处理

无。

约束说明

对于同一个算子,注册的算子输入名称需保持唯一,不能重复。

TensorType类说明

TensorType类用以定义输入或者输出支持的数据类型,TensorType提供以下接口指定支持的数据类型:

  • TensorType(DataType dt):指定仅支持一个数据类型;
  • TensorType(std::initializer_list<DataType> types):指定支持多个数据类型;
  • static TensorType ALL():指定支持所有数据类型;
  • static TensorType FLOAT():指定支持DT_FLOAT和DT_FLOAT16数据类型。
分享:

    相关文档

    相关产品

文档是否有解决您的问题?

提交成功!非常感谢您的反馈,我们会继续努力做到更好!
反馈提交失败,请稍后再试!

*必选

请至少选择或填写一项反馈信息

字符长度不能超过200

提交反馈 取消

如您有其它疑问,您也可以通过华为云社区问答频道来与我们联系探讨

智能客服提问云社区提问