ai.djl.mxnet.engine

Class DeviceType

java.lang.Object

ai.djl.mxnet.engine.DeviceType

public final class DeviceType
extends java.lang.Object

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.

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.

Parameters:

deviceType - the MXNet device number

Returns:

the corresponding Device