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

DYNAMIC_OUTPUT

函数原型

DYNAMIC_OUTPUT (x, t)

功能说明

注册动态算子输出信息。

注册动态算子输出信息成功后,自动生成动态算子输出的相关接口,包括用于创建动态输出、设置算子输出的对应描述等

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

_THIS_TYPE& create_dynamic_output_d(unsigned int num); // 创建动态输出d,包括num个输出
TensorDesc get_dynamic_output_desc_d(unsigned int index);// 返回动态输出d第index个描述,包括Shape、DataType、Format等信息
graphStatus update_dynamic_output_desc_d(unsigned int index, const TensorDesc& tensorDesc);// 更新动态输出d的第index个描述

参数说明

参数名

输入/输出

类型

描述

x

输入

-

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

t

输入

-

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

TensorType{DT_FLOAT}

TensorType({DT_FLOAT, DT_INT8}

返回值

无。

异常处理

无。

约束说明

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

分享:

    相关文档

    相关产品