简介
节点输出表达类,用于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 |
父主题: EsTensorHolder