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.