mindspore_lite.CPUDeviceInfo

class mindspore_lite.CPUDeviceInfo(enable_fp16=False)[source]

Helper class to set cpu device info, and it inherits DeviceInfo base class.

Parameters

enable_fp16 (bool, optional) – enables to perform the float16 inference. Default: False.

Raises

TypeErrorenable_fp16 is not a bool.

Examples

>>> 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: 0,
thread_affinity_core_list: [],
enable_parallel: False,
device_list: 0, .