MSConfig

import com.mindspore.lite.config.MSConfig;

MSConfig类用于保存执行中的配置变量。

公有成员函数

init

public boolean init(int deviceType, int threadNum, int cpuBindMode, boolean enable_float16)

初始化MSConfig。

• 参数

  • deviceType: 设备类型，deviceType在com.mindspore.lite.config.DeviceType中定义。
  • threadNum: 线程数。
  • cpuBindMode: CPU绑定模式，cpuBindMode在com.mindspore.lite.config.CpuBindMode中定义。
  • enable_float16：是否优先使用float16算子。

• 返回值

  初始化是否成功。

public boolean init(int deviceType, int threadNum, int cpuBindMode)

初始化MSConfig，enable_float16默认为false。

• 参数

  • deviceType: 设备类型，deviceType在com.mindspore.lite.config.DeviceType中定义。
  • threadNum: 线程数。
  • cpuBindMode: CPU绑定模式，cpuBindMode在com.mindspore.lite.config.CpuBindMode中定义。

• 返回值

  初始化是否成功。

public boolean init(int deviceType, int threadNum)

初始化MSConfig，cpuBindMode默认为CpuBindMode.MID_CPU，enable_float16默认为false。

• 参数

  • deviceType: 设备类型，deviceType在com.mindspore.lite.config.DeviceType中定义。
  • threadNum: 线程数。

• 返回值

  初始化是否成功。

public boolean init(int deviceType)

初始化MSConfig，cpuBindMode默认为CpuBindMode.MID_CPU，threadNum默认为2，enable_float16默认为false。

• 参数

  • deviceType: 设备类型，deviceType在com.mindspore.lite.config.DeviceType中定义。

• 返回值

  初始化是否成功。

public boolean init()

初始化MSConfig，deviceType默认为DeviceType.DT_CPU，cpuBindMode默认为CpuBindMode.MID_CPU，threadNum默认为2，enable_float16默认为false。

• 返回值

  初始化是否成功。

free

public void free()

释放MSConfig运行过程中动态分配的内存。LiteSession init之后即可释放。