torch.DeviceType
See theDeviceType companion object
enum DeviceType
Attributes
- Companion
- object
- Source
- Device.scala
- Graph
-
- Supertypes
Members list
Type members
Enum entries
case case COMPILE_TIME_MAX_DEVICE_TYPES extends DeviceType
Attributes
- Source
- Device.scala
case case CPU extends DeviceType
Attributes
- Source
- Device.scala
case case CUDA extends DeviceType
Attributes
- Source
- Device.scala
case case FPGA extends DeviceType
Attributes
- Source
- Device.scala
case case HIP extends DeviceType
Attributes
- Source
- Device.scala
case case HPU extends DeviceType
Attributes
- Source
- Device.scala
case case IDEEP extends DeviceType
Attributes
- Source
- Device.scala
case case IPU extends DeviceType
Attributes
- Source
- Device.scala
case case Lazy extends DeviceType
Attributes
- Source
- Device.scala
case case MKLDNN extends DeviceType
Attributes
- Source
- Device.scala
case case MPS extends DeviceType
Attributes
- Source
- Device.scala
case case MTIA extends DeviceType
Attributes
- Source
- Device.scala
case case Meta extends DeviceType
Attributes
- Source
- Device.scala
case case Metal extends DeviceType
Attributes
- Source
- Device.scala
case case OPENCL extends DeviceType
Attributes
- Source
- Device.scala
case case OPENGL extends DeviceType
Attributes
- Source
- Device.scala
case case ORT extends DeviceType
Attributes
- Source
- Device.scala
case case PrivateUse1 extends DeviceType
Attributes
- Source
- Device.scala
case case VE extends DeviceType
Attributes
- Source
- Device.scala
case case Vulkan extends DeviceType
Attributes
- Source
- Device.scala
case case XLA extends DeviceType
Attributes
- Source
- Device.scala
case case XPU extends DeviceType
Attributes
- Source
- Device.scala
In this article