Package | Description |
---|---|
org.bytedeco.pytorch |
Modifier and Type | Method and Description |
---|---|
KernelFunction |
KernelFunction.getPointer(long i) |
static KernelFunction |
KernelFunction.makeAmbiguousAutogradOther() |
static KernelFunction |
KernelFunction.makeFallthrough()
Create a KernelFunction from an unboxed function.
|
static KernelFunction |
KernelFunction.makeNamedNotSupported() |
KernelFunction |
KernelFunction.position(long position) |
Modifier and Type | Method and Description |
---|---|
RegistrationHandleRAII |
Dispatcher.registerFallback(short dispatch_key,
KernelFunction kernel,
String debug) |
RegistrationHandleRAII |
Dispatcher.registerFallback(torch.DispatchKey dispatch_key,
KernelFunction kernel,
BytePointer debug)
Register a fallback kernel for a backend.
|
RegistrationHandleRAII |
Dispatcher.registerImpl(OperatorName op_name,
DispatchKeyOptional dispatch_key,
KernelFunction kernel,
CppSignatureOptional cpp_signature,
FunctionSchema inferred_function_schema,
BytePointer debug)
Register a kernel to the dispatch table for an operator.
|
RegistrationHandleRAII |
Dispatcher.registerImpl(OperatorName op_name,
DispatchKeyOptional dispatch_key,
KernelFunction kernel,
CppSignatureOptional cpp_signature,
FunctionSchema inferred_function_schema,
String debug) |
Copyright © 2024. All rights reserved.