FrameworkType

函数功能

设置原始模型的框架类型。

函数原型

[OpRegistrationData](OpRegistrationData.md) &FrameworkType(const domi::FrameworkType &fmk_type)

参数说明

参数

输入/输出

说明

fmk_type

输入

框架类型:

  • CAFFE
  • TENSORFLOW
  • ONNX

FrameworkType枚举值的定义如下:

enum FrameworkType {
CAFFE = 0,
MINDSPORE = 1,
TENSORFLOW = 3,
ANDROID_NN,
ONNX,
FRAMEWORK_RESERVED,
};