Class AutoDeviceInfo

Inheritance Relationships

Base Type

Class Documentation

class AutoDeviceInfo : public mindspore::DeviceInfoContext

Derived from DeviceInfoContext, The configuration of the model running auto on the Host Devices, include CPU/GPU/NPU/Ascend. 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.