GetAllFeatureMemoryTypeSize

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

Obtains the feature memory sizes of all memory types after compilation. Currently, only the fixed memory is included (applicable to the graphs of static and dynamic shapes).

Prototype

std::vector<FeatureMemoryPtr> GetAllFeatureMemoryTypeSize() const

Parameters

None

Returns

shared_ptr of FeatureMemory in the graph compilation result.

Constraints

None

Parent topic: CompiledGraphSummary