aclgrphParseCaffe

【免费下载链接】ge GE(Graph Engine)是面向昇腾的图编译器和执行器,提供了计算图优化、多流并行、内存复用和模型下沉等技术手段,加速模型执行效率,减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力,并同时支持 onnx、pb 等主流模型格式的解析与编译。 【免费下载链接】ge 项目地址: https://gitcode.com/cann/ge

产品支持情况

  • Ascend 950PR/Ascend 950DT:不支持
  • Atlas A3 训练系列产品/Atlas A3 推理系列产品:不支持
  • Atlas A2 训练系列产品/Atlas A2 推理系列产品:不支持
  • Atlas 200I/500 A2 推理产品:支持
  • Atlas 推理系列产品:支持
  • Atlas 训练系列产品:支持
  • IPV350:不支持

头文件/库文件

  • 头文件:#include <parser/caffe_parser.h>
  • 库文件:libfmk_parser.so

功能说明

将Caffe模型解析为图。

函数原型

graphStatus aclgrphParseCaffe(const char *model_file, const char *weights_file, ge::Graph &graph)
graphStatus aclgrphParseCaffe(const char *model_file, const char *weights_file, const std::map<ge::AscendString, ge::AscendString> &parser_params, ge::Graph &graph)

参数说明

参数名 输入/输出 说明
model_file 输入 Caffe原始模型文件路径。
weights_file 输入 Caffe权重文件路径。
parser_params 输入 配置参数map映射表,key为参数类型,value为参数值,均为AscendString格式,用于描述原始模型解析参数。
map中支持的配置参数请参见Parser解析接口支持的配置参数
graph 输出 解析后生成的图。

返回值说明

参数名 类型 说明
- graphStatus GRAPH_SUCCESS(0):成功。
其他值:失败。

约束说明

【免费下载链接】ge GE(Graph Engine)是面向昇腾的图编译器和执行器,提供了计算图优化、多流并行、内存复用和模型下沉等技术手段,加速模型执行效率,减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力,并同时支持 onnx、pb 等主流模型格式的解析与编译。 【免费下载链接】ge 项目地址: https://gitcode.com/cann/ge

Logo

Agent 垂直技术社区,欢迎活跃、内容共建。

更多推荐