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

GetCapturedTensor

产品支持情况

产品

是否支持

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

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

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

头文件/库文件

  • 头文件:/include/ge/fusion/match_result.h
  • 库文件:libge_compiler.so

功能说明

根据pattern中捕获的索引,获取匹配中对应的node output。

函数原型

1
Status GetCapturedTensor(size_t capture_idx, NodeIo &node_output) const

参数说明

参数名

输入/输出

说明

capture_idx

输入

索引ID。

node_output

输出

配到的tensor。

返回值说明

参数名

类型

说明

-

Status

SUCCESS:成功。

FAILED:失败 。

约束说明