简介
ComputeNodeInfo用于序列化保存算子的相关编译信息,包括算子类型、算子名称、输入输出数量、编译时张量的基本信息(CompileTimeTensorDesc)和输入输出实例化信息(AnchorInstanceInfo)。此外,该类还支持获取和设置算子原型中定义的属性的值(RuntimeAttrs)。ComputeNodeInfo的序列化内存空间是平铺式的,内存排布顺序为ComputeNodeInfo数据成员、算子输入输出实例化信息、编译时张量的基本信息(数量与实际输入、输出个数相同)、算子属性信息。
需要包含的头文件
1 | #include <compute_node_info.h> |
Public成员函数
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)
父主题: ComputeNodeInfo