GetAllFeatureMemoryTypeSize

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

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

Prototype

std::vector<FeatureMemoryPtr> GetAllFeatureMemoryTypeSize() const

Parameters

None

Returns

shared_ptr of FeatureMemory in the graph compilation result.

Restrictions

None

Parent topic: CompiledGraphSummary