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

GetCapturedTensors

产品支持情况

产品

是否支持

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

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

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

头文件/库文件

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

功能说明

获取pattern中捕获的所有tensor,vector中顺序为捕获顺序。

函数原型

1
Status GetCapturedTensors(std::vector<NodeIo> &node_outputs) const

参数说明

参数名

输入/输出

说明

node_outputs

输入

捕获pattern中的所有tensor。

返回值说明

参数名

类型

说明

-

Status

SUCCESS:获取成功。

FAILED:获取失败 。

约束说明