Class CPUDeviceInfo
Defined in File context.h
Inheritance Relationships
Base Type
public mindspore::DeviceInfoContext(Class DeviceInfoContext)
Class Documentation
-
class CPUDeviceInfo : public mindspore::DeviceInfoContext
Derived from DeviceInfoContext, The configuration of the model running on the CPU. This option is only valid for MindSpore Lite.
Public Functions
-
inline virtual enum DeviceType GetDeviceType() const override
Get the type of this DeviceInfoContext.
- Returns
Type of this DeviceInfoContext.
-
void SetEnableFP16(bool is_fp16)
Set enables to perform the float16 inference.
- Parameters
is_fp16 – [in] Enable float16 inference or not.
-
bool GetEnableFP16() const
Get enables to perform the float16 inference.
- Returns
Whether enable float16 inference.
-
inline virtual enum DeviceType GetDeviceType() const override