MSConfig
import com.mindspore.lite.config.MSConfig;
MSConfig类用于保存执行中的配置变量。
公有成员函数
function |
|---|
boolean init(int deviceType, int threadNum, int cpuBindMode, boolean enable_float16) |
boolean init(int deviceType, int threadNum, int cpuBindMode) |
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之后即可释放。