Class KirinNPUDeviceInfo

• Defined in File context.h

Inheritance Relationships

Base Type

• public mindspore::DeviceInfoContext (Class DeviceInfoContext)

Class Documentation

class KirinNPUDeviceInfo : public mindspore::DeviceInfoContext

Derived from DeviceInfoContext, The configuration of the model running on the NPU. This option is only valid for MindSpore Lite.

Public Functions

inline virtual enum DeviceType GetDeviceType() const override

Get the type of this DeviceInfoContext.

返回

Type of this DeviceInfoContext.

void SetEnableFP16(bool is_fp16)

Set enables to perform the float16 inference.

参数

is_fp16 – [in] Enable float16 inference or not.

bool GetEnableFP16() const

Get enables to perform the float16 inference.

返回

Whether enable float16 inference.

void SetFrequency(int frequency)

Set the NPU frequency.

参数

frequency – [in] Can be set to 1 (low power consumption), 2 (balanced), 3 (high performance), 4 (extreme performance), default as 3.

int GetFrequency() const

Get the NPU frequency.

返回

NPU frequency