| Package | Description |
|---|---|
| org.bytedeco.pytorch | |
| org.bytedeco.pytorch.global |
| Modifier and Type | Method and Description |
|---|---|
void |
OptionalDeviceGuard.reset_device(Device device,
DeviceGuardImplInterface impl)
For testing only
|
| Constructor and Description |
|---|
DeviceGuardImplRegistrar(byte arg0,
DeviceGuardImplInterface arg1) |
DeviceGuardImplRegistrar(torch.DeviceType arg0,
DeviceGuardImplInterface arg1) |
OptionalDeviceGuard(Device device,
DeviceGuardImplInterface impl)
Constructor for testing only.
|
| Modifier and Type | Method and Description |
|---|---|
static DeviceGuardImplInterface |
torch.getDeviceGuardImpl(byte type) |
static DeviceGuardImplInterface |
torch.getDeviceGuardImpl(torch.DeviceType type) |
Copyright © 2024. All rights reserved.