GEStreamAllocationSummaryGetAllNodes

Applicability

Product

Supported or Not

Atlas A3 training products/Atlas A3 inference products

Atlas A2 training products/Atlas A2 inference products

Atlas 200I/500 A2 inference products

x

Atlas inference products

Atlas training products

Header File/Library File

  • Header file: #include <ge/ge_graph_compile_summary.h>
  • Library file: libge_compiler.so

Function Usage

Obtains all nodes of the root graph and subgraphs.

Prototype

1
ge::Status GEStreamAllocationSummaryGetAllNodes(const ge::CompiledGraphSummary &compiled_graph_summary, std::map<AscendString, std::vector<std::vector<GNode>>> &graph_to_all_nodes);

Parameters

Parameter

Input/Output

Description

compiled_graph_summary

Input

Brief information about the graph build result.

graph_to_all_nodes

Output

A value in Map format. The key is the graph name. The value is the vector indicating all nodes. The index indicates the logical stream ID.

Returns

Parameter

Type

Description

-

ge::Status

  • SUCCESS: The API is called successfully.
  • FAILED: The API fails to be called.

Constraints

None