昇腾社区首页
中文
注册
开发者
下载

IONum

函数功能

设置算子输入输出IR原型个数,用于构造各子类Context的基类ExtendedKernelContext中ComputeNodeInfo信息。该接口默认每个IR原型输入输出的实例个数为1。

函数原型

1
T &IONum(size_t input_ir_num, size_t output_ir_num)

参数说明

参数

输入/输出

说明

input_ir_num

输入

输入IR原型个数。

output_ir_num

输入

输出IR原型个数。

返回值说明

返回子类对象T类型的引用,用于支持子类链式调用。

约束说明

此接口与IOInstanceNum接口互斥。仅需调用2种接口的一种即可,以先调用的接口为准。