昇腾社区首页
中文
注册

简介

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)