Overview
ComputeNodeInfo is used to serialize and save the compilation information of an operator, including the operator type, operator name, number of inputs and outputs, basic information about the tensor during compilation (CompileTimeTensorDesc), and input and output instantiation information (AnchorInstanceInfo). In addition, this class can be used to obtain and set the values of attributes defined in the operator prototype (RuntimeAttrs). The serialization memory space of ComputeNodeInfo is tiled. The memory layout order is as follows: ComputeNodeInfo data members, operator input and output instantiation information, basic information about tensors during compilation (the number of tensors is the same as the actual number of inputs and outputs), and operator attribute information.
Header Files to Be Included
1 | #include <compute_node_info.h> |
Public Member Functions
ComputeNodeInfo() = delete ComputeNodeInfo(const ComputeNodeInfo &) = delete ComputeNodeInfo(ComputeNodeInfo &&) = delete ComputeNodeInfo &operator=(const ComputeNodeInfo &) = delete ComputeNodeInfo &operator=(ComputeNodeInfo &&) = delete const ge::char_t *GetNodeType() const const ge::char_t *GetNodeName() const size_t GetIrInputsNum() const size_t GetIrOutputsNum() const size_t GetInputsNum() const size_t GetOutputsNum() const const AnchorInstanceInfo *GetInputInstanceInfo(const size_t ir_index) const const AnchorInstanceInfo *GetOutputInstanceInfo(const size_t ir_index) const const CompileTimeTensorDesc *GetInputTdInfo(const size_t index) const const CompileTimeTensorDesc *GetOutputTdInfo(const size_t index) const const RuntimeAttrs *GetAttrs() const void SetNodeType(const ge::char_t *node_type) void SetNodeName(const ge::char_t *node_name) AnchorInstanceInfo *MutableInputInstanceInfo(const size_t ir_index) const AnchorInstanceInfo *MutableOutputInstanceInfo(const size_t ir_index) const CompileTimeTensorDesc *MutableInputTdInfo(const size_t index) const CompileTimeTensorDesc *MutableOutputTdInfo(const size_t index) const RuntimeAttrs *MutableAttrs() const void Init(const size_t ir_inputs_num, const size_t inputs_num, const size_t outputs_num, const ge::char_t *node_name, const ge::char_t *node_type) void Init(const size_t ir_inputs_num, const size_t ir_outputs_num, const size_t inputs_num, const size_t outputs_num, const size_t attr_size, const ge::char_t *node_name, const ge::char_t *node_type) static ge::graphStatus CalcSize(const size_t ir_inputs_num, const size_t inputs_num, const size_t outputs_num, size_t &total_size) static ge::graphStatus CalcSize(const size_t ir_inputs_num, const size_t ir_outputs_num, const size_t inputs_num, const size_t outputs_num, size_t &total_size)