Overview
Node output expression class. This class is used for C++ graph building. This class provides the tensor operation APIs of C and provides various operations on node outputs, including setting the data type, format, and shape, setting private attributes, and performing arithmetic operations.
Header Files to Be Included
#include "es_tensor_holder.h"
Public Member Functions
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 |
Parent topic: EsTensorHolder