昇腾社区首页
中文
注册

InsertOutputs

产品支持情况

产品

是否支持

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

Atlas 200/300/500 推理产品

函数功能

在FusionScopesResult中插入融合算子的输出信息。

函数原型

数据类型为string的接口后续版本会废弃,建议使用数据类型为非string的接口。

1
2
void InsertOutputs(const std::string &inner_op_name, const std::vector<int32_t> &index_map);
void InsertOutputs(const char_t *inner_op_name, const std::vector<int32_t> &index_map);

参数说明

参数名

输入/输出

描述

inner_op_name

输入

融合算子名称

index_map

输入

融合算子index信息,是一个vector类型。vector的index表示scope内部小算子的输入index,具体值表示融合算子的输入index,如果融合算子没有使用这个index,则使用占位符kFusionDisableIndex表示。

返回值

无。

约束说明

无。