Package | Description |
---|---|
org.bytedeco.pytorch | |
org.bytedeco.pytorch.cuda | |
org.bytedeco.pytorch.functions | |
org.bytedeco.pytorch.global |
Modifier and Type | Method and Description |
---|---|
Allocator |
StorageImpl.allocator() |
Allocator |
Storage.allocator() |
Allocator |
CUDAHooksInterface.getCUDADeviceAllocator() |
Allocator |
MPSHooksInterface.getMPSDeviceAllocator() |
Allocator |
HIPHooksInterface.getPinnedMemoryAllocator() |
Allocator |
CUDAHooksInterface.getPinnedMemoryAllocator() |
Modifier and Type | Method and Description |
---|---|
TensorMaker |
TensorMaker.allocator(Allocator allocator) |
void |
StorageImpl.set_allocator(Allocator allocator) |
Constructor and Description |
---|
Storage(Storage.use_byte_size_t arg0,
long size_bytes,
DataPtr data_ptr,
Allocator allocator,
boolean resizable) |
Storage(Storage.use_byte_size_t arg0,
SymInt size_bytes,
Allocator allocator,
boolean resizable) |
StorageImpl(StorageImpl.use_byte_size_t arg0,
SymInt size_bytes,
Allocator allocator,
boolean resizable) |
StorageImpl(StorageImpl.use_byte_size_t arg0,
SymInt size_bytes,
DataPtr data_ptr,
Allocator allocator,
boolean resizable) |
Modifier and Type | Class and Description |
---|---|
class |
CUDAAllocator |
Modifier and Type | Method and Description |
---|---|
StorageImplPtr |
StorageImplCreateHelper.call(StorageImpl.use_byte_size_t arg0,
SymInt size_bytes,
Allocator allocator,
boolean resizable) |
Modifier and Type | Method and Description |
---|---|
static Allocator |
torch.GetAllocator(byte t) |
static Allocator |
torch.GetAllocator(torch.DeviceType t) |
static Allocator |
torch.getCPUAllocator() |
static Allocator |
torch_cuda.getCUDADeviceAllocator() |
Modifier and Type | Method and Description |
---|---|
static TensorBase |
torch.empty_generic(long[] size,
Allocator allocator,
DispatchKeySet ks,
torch.ScalarType scalar_type,
MemoryFormatOptional memory_format_opt) |
static TensorBase |
torch.empty_generic(LongArrayRef size,
Allocator allocator,
DispatchKeySet ks,
torch.ScalarType scalar_type,
MemoryFormatOptional memory_format_opt) |
static TensorBase |
torch.empty_strided_generic(long[] size,
long[] stride,
Allocator allocator,
DispatchKeySet ks,
torch.ScalarType scalar_type) |
static TensorBase |
torch.empty_strided_generic(LongArrayRef size,
LongArrayRef stride,
Allocator allocator,
DispatchKeySet ks,
torch.ScalarType scalar_type) |
static TensorBase |
torch.empty_strided_symint_generic(SymIntArrayRef size,
SymIntArrayRef stride,
Allocator allocator,
DispatchKeySet ks,
torch.ScalarType scalar_type) |
static void |
torch.SetAllocator(byte t,
Allocator alloc) |
static void |
torch.SetAllocator(byte t,
Allocator alloc,
byte priority) |
static void |
torch.SetAllocator(torch.DeviceType t,
Allocator alloc) |
static void |
torch.SetAllocator(torch.DeviceType t,
Allocator alloc,
byte priority)
Set the allocator for DeviceType
t . |
Copyright © 2024. All rights reserved.