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

GetInputConstData

产品支持情况

产品

是否支持

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

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

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

头文件和库文件

  • 头文件:#include <graph/operator.h>
  • 库文件:libgraph.so

功能说明

如果指定算子Input对应的节点为Const节点,可调用该接口获取Const节点的数据。

函数原型

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

1
2
graphStatus GetInputConstData(const std::string &dst_name, Tensor &data) const
graphStatus GetInputConstData(const char_t *dst_name, Tensor &data) const

参数说明

参数名

输入/输出

描述

dst_name

输入

输入名称。

data

输出

返回Const节点的数据Tensor。

返回值说明

graphStatus类型:如果指定算子Input对应的节点为Const节点且获取数据成功,返回GRAPH_SUCCESS,否则,返回GRAPH_FAILED。

异常处理

无。

约束说明

无。