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

简介

节点输出表达类,用于C++构图,封装了C的张量操作接口,该类提供了对节点输出的各种操作,包括设置dataype、format、shape等信息,以及设置私有属性,算术运算操作等能力。

需要包含的头文件

#include "es_tensor_holder.h"

Public成员函数

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
EsTensorHolder operator+(const EsTensorHolder &other) const
EsTensorHolder operator-(const EsTensorHolder &other) const
EsTensorHolder operator*(const EsTensorHolder &other) const
EsTensorHolder operator/(const EsTensorHolder &other) const
EsTensorHolder(EsCTensorHolder *tensor) : tensor_holder_(tensor) {}
EsTensorHolder() = default;
EsTensorHolder &SetDataType(const ge::DataType data_type)
EsTensorHolder &SetFormat(const ge::Format format)
EsTensorHolder &SetShape(const std::vector<int64_t> &dims)
EsTensorHolder &SetOriginSymbolShape(const std::vector<const char *> &dims)
Status SetAttr(const char *attr_name, int64_t value)
Status SetAttr(const char *attr_name, const char *value)
Status SetAttr(const char *attr_name, bool value)
Status SetAttrForNode(const char *attr_name, int64_t value)
Status SetAttrForNode(const char *attr_name, const char *value)
Status SetAttrForNode(const char *attr_name, bool value)
EsCTensorHolder *GetCTensorHolder() const
int32_t GetProducerOutIndex() const
GNode *GetProducer
Status AddControlEdge(const std::vector<EsTensorHolder> &ctrl_ins) const