Package smile.deep.tensor

Enum Class DeviceType

java.lang.Object

java.lang.Enum<DeviceType>

smile.deep.tensor.DeviceType

All Implemented Interfaces:

Serializable, Comparable<DeviceType>, Constable

public enum DeviceType
extends Enum<DeviceType>

The compute device type.

Nested Class Summary

Nested classes/interfaces inherited from class java.lang.Enum

Enum.EnumDesc<E extends Enum<E>>

Enum Constant Summary

Enum Constants

Enum Constant	Description
CPU	CPU
CUDA	NVIDIA GPU
MPS	GPU for macOS devices with Metal programming framework.

Method Summary

Modifier and Type	Method	Description
byte	value()	Returns the byte value of device type, which is compatible with PyTorch.
static DeviceType	valueOf(String name)	Returns the enum constant of this class with the specified name.
static DeviceType[]	values()	Returns an array containing the constants of this enum class, in the order they are declared.

Methods inherited from class java.lang.Enum

clone, compareTo, describeConstable, equals, finalize, getDeclaringClass, hashCode, name, ordinal, toString, valueOf

Methods inherited from class java.lang.Object

getClass, notify, notifyAll, wait, wait, wait

Enum Constant Details

CPU

public static final DeviceType CPU

CPU

CUDA

public static final DeviceType CUDA

NVIDIA GPU

MPS

public static final DeviceType MPS

GPU for macOS devices with Metal programming framework.

Method Details

values

public static DeviceType[] values()

Returns an array containing the constants of this enum class, in the order they are declared.

Returns:

an array containing the constants of this enum class, in the order they are declared

valueOf

public static DeviceType valueOf(String name)

Returns the enum constant of this class with the specified name. The string must match exactly an identifier used to declare an enum constant in this class. (Extraneous whitespace characters are not permitted.)

Parameters:

name - the name of the enum constant to be returned.

Returns:

the enum constant with the specified name

Throws:

IllegalArgumentException - if this enum class has no constant with the specified name

NullPointerException - if the argument is null

value

public byte value()

Returns the byte value of device type, which is compatible with PyTorch.

Returns:

the byte value of device type.