更新时间:2021-03-18 GMT+08:00
分享

OPTIONAL_INPUT

函数原型

OPTIONAL_INPUT(x, t)

功能说明

注册可选算子输入信息。

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

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

static const string name_in_b(); // 返回输入的名称,即“b”
_THIS_TYPE& set_input_b(Operator& v, const string& srcName);// 指定输入b与算子对象v的输出srcName存在连接关系,返回算子对象本身
_THIS_TYPE& set_input_b(Operator& v); // 指定输入b与算子对象v的索引0的输出存在连接关系,返回算子对象本身
TensorDesc get_input_desc_b(); // 返回输入b对应的描述
graphStatus update_input_desc_b(const TensorDesc& tensorDesc);// 设置输入b对应的描述,包括Shape、DataType、Format等信息

参数说明

参数名

输入/输出

类型

描述

x

输入

-

宏参数,算子输入的名称。

t

输入

-

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

TensorType{DT_FLOAT}

TensorType({DT_FLOAT, DT_INT8}

返回值

无。

异常处理

无。

约束说明

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

分享:

    相关文档

    相关产品