DYNAMIC_INPUT
函数原型
DYNAMIC_INPUT (x, t)
功能说明
注册动态算子输入信息。
注册动态算子输入信息成功后,自动生成算子输入的相关接口,用于创建动态输入、设置算子输入的对应描述等。
例如,注册动态算子输入d,算子输入接收的数据类型为TensorType{DT_FLOAT},可调用DYNAMIC_INPUT(d, TensorType{DT_FLOAT})接口,注册动态算子输入成功后,自动生成以下相关接口:
_THIS_TYPE& create_dynamic_input_d(unsigned int num); // 创建动态输入d,包括num个输入 TensorDesc get_dynamic_input_desc_d(unsigned int index);// 返回动态输入d第index个描述,包括Shape、DataType、Format等信息 graphStatus update_dynamic_input_desc_d(unsigned int index, const TensorDesc& tensorDesc);// 更新动态输入d的第index个描述 _THIS_TYPE& set_dynamic_input_d(unsigned int dstIndex, Operator &v); // 指定输入d的第dstIndex个输入与算子对象v的索引0的输出存在连接关系,返回算子对象本身 _THIS_TYPE& set_dynamic_input_d(unsigned int dstIndex, Operator &v, const string &srcName); / /指定动态输入d的第dstIndex个输入与算子对象v的输出srcName存在连接关系,返回算子对象本身
参数说明
参数名 |
输入/输出 |
类型 |
描述 |
---|---|---|---|
x |
输入 |
- |
宏参数,算子输入的名称 |
t |
输入 |
- |
算子输入接收的数据类型,可以是TensorType定义的一个或多个,如果多个,通过“,”隔离,例如: TensorType{DT_FLOAT} TensorType({DT_FLOAT, DT_INT8} |
返回值
无。
异常处理
无。
约束说明
对于同一个算子,注册的算子输入名称需保持唯一,不能重复。