Init

Function Usage

Initializes the ComputeNodeInfo class.

Prototype

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)

Parameters

Parameter	Input/Output	Description
ir_inputs_num	Input	Number of operator prototype inputs.
inputs_num	Input	Actual number of inputs of an operator.
outputs_num	Input	Actual number of outputs of an operator.
node_name	Input	Node name.
node_type	Input	Node type (operator prototype name).
ir_outputs_num	Input	Number of operator prototype outputs.
attr_size	Input	Number of attributes.

Returns

None

Constraints

None

Examples

auto ir_input_num = compute_node->GetOpDesc()->GetIrInputs().size();
auto inputs_num = compute_node->GetInDataNodesAndAnchors().size();
auto outputs_num = compute_node->GetOutDataNodesAndAnchors().size();
auto compute_node_info_holder = std::make_unique<uint8_t[]>(total_size);
auto compute_node_info = reinterpret_cast<ComputeNodeInfo *>(compute_node_info_holder.get());
compute_node_info->Init(ir_input_num, inputs_num, outputs_num, node_name, node_type);

Parent topic: ComputeNodeInfo