昇腾社区首页
中文
注册
开发者
下载

GetAllFeatureMemoryTypeSize

产品支持情况

产品

是否支持

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品/Atlas A2 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

头文件/库文件

  • 头文件:/include/ge/ge_graph_compile_summary.h
  • 库文件:libge_compiler.so

功能说明

获取编译后所有内存类型的Feature内存大小,当前仅包含Fixed的内存(适用于静态shape图和动态shape图)。

函数原型

1
std::vector<FeatureMemoryPtr> GetAllFeatureMemoryTypeSize() const

参数说明

返回值说明

图编译结果中FeatureMemory的shared_ptr。FeatureMemory详细结构请参见FeatureMemory

约束说明