FrameworkType

Description

Sets the framework type of the original model.

Prototype

OpRegistrationData &FrameworkType(const domi::FrameworkType &fmk_type)

Parameters

Parameter

Input/Output

Description

fmk_type

Input

Framework type.

  • CAFFE
  • TENSORFLOW
  • ONNX

The enum values of FrameworkType are defined as follows:

enum FrameworkType {

CAFFE = 0,

MINDSPORE = 1,

TENSORFLOW = 3,

ANDROID_NN,

ONNX,

FRAMEWORK_RESERVED,

};