GetOutputShapes

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	√
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

Applies only to static shape graphs to obtain the output shape for graph build. The output shape can be used to calculate the memory size occupied by outputs. For dynamic multi-profile graphs, it obtains the output shape 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.

Constraints

None

Parent topic: CompiledGraphSummary