跳到主要内容

Init

函数功能

初始化ComputeNodeInfo类。

函数原型

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);

参数说明

参数输入/输出说明
ir_inputs_num输入算子原型输入的个数。
inputs_num输入算子实际输入个数。
outputs_num输入算子实际输出个数。
node_name输入节点名称。
node_type输入节点类型(算子原型名称)。
ir_outputs_num输入算子原型输出的个数。
attr_size输入属性个数。

返回值

约束说明

调用示例

auto ir_input_num = node->GetOpDesc()->GetIrInputs().size();
auto ir_output_num = node->GetOpDesc()->GetIrOutputs().size();
auto inputs_num = node->GetInDataNodesAndAnchors().size();
auto outputs_num = node->GetOutDataNodesAndAnchors().size();
GE_ASSERT_SUCCESS(ComputeNodeInfo::Init(ir_input_num, inputs_num,outputs_num, node_name, node_type));