ai.djl.mxnet.engine

Class MxDeviceType

java.lang.Object

ai.djl.mxnet.engine.MxDeviceType

All Implemented Interfaces:

ai.djl.DeviceType

public final class MxDeviceType
extends java.lang.Object
implements ai.djl.DeviceType

DeviceType is the MXNet equivalent of the types in Device.

Method Summary

Modifier and Type	Method and Description
static java.lang.String	fromDeviceType(int deviceType) Converts from an MXNet device number to Device.
static int	toDeviceType(ai.djl.Device device) Converts a Device to the corresponding MXNet device number.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

toDeviceType

public static int toDeviceType(ai.djl.Device device)

Converts a Device to the corresponding MXNet device number.

Specified by:

toDeviceType in interface ai.djl.DeviceType

Parameters:

device - the java Device

Returns:

the MXNet device number

fromDeviceType

public static java.lang.String fromDeviceType(int deviceType)

Converts from an MXNet device number to Device.

Specified by:

fromDeviceType in interface ai.djl.DeviceType

Parameters:

deviceType - the MXNet device number

Returns:

the corresponding Device