| Package | Description |
|---|---|
| org.bytedeco.pytorch | |
| org.bytedeco.pytorch.cuda | |
| org.bytedeco.pytorch.global |
| Modifier and Type | Method and Description |
|---|---|
torch.DeviceType |
StorageImpl.device_type() |
torch.DeviceType |
StreamData3.device_type() |
torch.DeviceType |
TensorIteratorBase.device_type() |
torch.DeviceType |
TensorImpl.device_type()
The device type of a Tensor, e.g., DeviceType::CPU or DeviceType::CUDA.
|
static torch.DeviceType |
CPUGeneratorImpl.device_type() |
torch.DeviceType |
Storage.device_type() |
torch.DeviceType |
Stream.device_type() |
torch.DeviceType |
TensorIteratorBase.device_type(int arg) |
torch.DeviceType |
DeviceTypeSet.front() |
torch.DeviceType |
DeviceTypeSet.Iterator.get() |
torch.DeviceType |
DeviceGuardImplInterface.type()
Return the type of device managed by this guard implementation.
|
torch.DeviceType |
Device.type()
Returns the type of device this is.
|
| Modifier and Type | Method and Description |
|---|---|
StreamData3 |
StreamData3.device_type(torch.DeviceType setter) |
void |
DeviceTypeSet.erase(torch.DeviceType value) |
Device |
Context.getDeviceFromPtr(Pointer data,
torch.DeviceType device_type) |
void |
DeviceTypeSet.insert(torch.DeviceType value) |
StreamOptional |
Node.stream(torch.DeviceType device_type)
Note: Function Streams
A function's stream (for a given device type) is the stream of the first
element of its input buffer on a device of that type.
|
static Stream |
Stream.unpack3(long stream_id,
byte device_index,
torch.DeviceType device_type) |
| Constructor and Description |
|---|
Device(torch.DeviceType type) |
Device(torch.DeviceType type,
byte index)
Constructs a new
Device from a DeviceType and an optional device
index. |
DeviceGuardImplRegistrar(torch.DeviceType arg0,
DeviceGuardImplInterface arg1) |
| Modifier and Type | Method and Description |
|---|---|
torch.DeviceType |
CUDAStream.device_type()
Used to avoid baking in device type explicitly to Python-side API.
|
| Modifier and Type | Method and Description |
|---|---|
static CUDAStream |
CUDAStream.unpack3(long stream_id,
byte device_index,
torch.DeviceType device_type) |
| Modifier and Type | Method and Description |
|---|---|
static torch.DeviceType |
torch.backendToDeviceType(torch.Backend b) |
static torch.DeviceType |
torch.dispatchKeyToDeviceType(torch.DispatchKey dispatch_key) |
torch.DeviceType |
torch.DeviceType.intern() |
static torch.DeviceType |
torch.kCPU() |
static torch.DeviceType |
torch.kCUDA() |
static torch.DeviceType |
torch.kFPGA() |
static torch.DeviceType |
torch.kHIP() |
static torch.DeviceType |
torch.kHPU() |
static torch.DeviceType |
torch.kIPU() |
static torch.DeviceType |
torch.kLazy() |
static torch.DeviceType |
torch.kMeta() |
static torch.DeviceType |
torch.kMetal() |
static torch.DeviceType |
torch.kMPS() |
static torch.DeviceType |
torch.kMTIA() |
static torch.DeviceType |
torch.kORT() |
static torch.DeviceType |
torch.kPrivateUse1() |
static torch.DeviceType |
torch.kVE() |
static torch.DeviceType |
torch.kVulkan() |
static torch.DeviceType |
torch.kXLA() |
static torch.DeviceType |
torch.kXPU() |
static torch.DeviceType |
torch.DeviceType.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static torch.DeviceType[] |
torch.DeviceType.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
Copyright © 2024. All rights reserved.