Class AutoDeviceInfo

• Defined in File context.h

Inheritance Relationships

Base Type

• public mindspore::DeviceInfoContext (Class DeviceInfoContext)

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.

Returns

Type of this DeviceInfoContext.