mindspore_lite.CPUDeviceInfo
- class mindspore_lite.CPUDeviceInfo(enable_fp16=False)[源代码]
用于设置CPU设备信息的Helper类,继承自DeviceInfo基类。
- 参数:
enable_fp16 (bool,可选) - 启用以执行float16推理。默认值:False。
- 异常:
TypeError - enable_fp16 不是bool类型。
样例:
>>> import mindspore_lite as mslite >>> cpu_device_info = mslite.CPUDeviceInfo(enable_fp16=True) >>> print(cpu_device_info) device_type: DeviceType.kCPU, enable_fp16: True. >>> context = mslite.Context() >>> context.append_device_info(cpu_device_info) >>> print(context) thread_num: 0, inter_op_parallel_num: 0, thread_affinity_mode: 1, thread_affinity_core_list: [], enable_parallel: False, device_list: 0, .