GetOutputShapes

Applicability

Product	Supported
Atlas 350 Accelerator Card	√
Atlas A3 training product/Atlas A3 inference product	√
Atlas A2 training product/Atlas A2 inference product	√
Atlas 200I/500 A2 inference product	√
Atlas inference product	√
Atlas training product	√

Header File/Library File

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

Function Usage

Applies only to static shape graphs. It obtains the output shapes of the compiled graph. The output shapes can be used to calculate the memory size occupied by outputs. For dynamic multi-profile graphs, it obtains the output shapes of the maximum profile.

Prototype

Status GetOutputShapes(std::vector<ge::Shape> &shapes) const

Parameters

Parameter	Input/Output	Description
shapes	Output	Vector of the ge::Shape type.

Returns

Parameter	Type	Description
-	Status	SUCCESS: The API is called successfully. FAILED: The API fails to be called.

Restrictions

None

Parent topic: CompiledGraphSummary