Package | Description |
---|---|
org.bytedeco.pytorch | |
org.bytedeco.pytorch.cuda | |
org.bytedeco.pytorch.global |
Class and Description |
---|
torch.ActivityType |
torch.AliasAnalysisKind |
torch.AttributeKind |
torch.Backend
This legacy enum class defines the set of backends supported by old school,
code generated Type-based ATen.
|
torch.BackendComponent |
torch.CreationMeta
NOTE [ Autograd View Variables ]
Many operations return Variable that shares storage with an input Variable.
|
torch.DebugInfoKind |
torch.DeviceType |
torch.DispatchKey |
torch.EventFlag
Flags defining the behavior of events.
|
torch.ExecutorExecutionMode |
torch.Float32MatmulPrecision |
torch.FusionBehavior |
torch.JitAttributeKind |
torch.Layout |
torch.LinalgBackend |
torch.MemoryFormat
Construct an ArrayRef from a C array.
|
torch.NameType |
torch.ProfilerState |
torch.QEngine
QEngine is an enum that is used to select the engine to run quantized ops.
|
torch.QScheme
QScheme is an enum that specifies the type of quantization.
|
torch.RecordScope |
torch.ScalarType |
torch.SchemaArgType |
torch.Tag
Include this file if you want to register operators.
|
torch.TorchFunctionDisabledState |
torch.TypeKind |
Class and Description |
---|
torch_cuda.RecordContext |
torch.DeviceType |
torch.MemoryFormat
Construct an ArrayRef from a C array.
|
Copyright © 2024. All rights reserved.