| Package | Description |
|---|---|
| org.bytedeco.pytorch |
| Modifier and Type | Method and Description |
|---|---|
DispatchKeyOptional |
DispatchKeyOptional.put(DispatchKeyOptional x) |
DispatchKeyOptional |
DispatchKeyOptional.put(torch.DispatchKey value) |
| Modifier and Type | Method and Description |
|---|---|
OperatorName |
Dispatcher.getRegistrationsForDispatchKey(DispatchKeyOptional k)
Useful for inspecting global Dispatcher registration state.
|
DispatchKeyOptional |
DispatchKeyOptional.put(DispatchKeyOptional x) |
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) |
void |
Dispatcher.waitForImpl(OperatorName op_name,
DispatchKeyOptional dispatch_key) |
| Constructor and Description |
|---|
Library(int kind,
String ns,
DispatchKeyOptional k,
String file,
int line) |
Library(Library.Kind kind,
BytePointer ns,
DispatchKeyOptional k,
BytePointer file,
int line)
\private
Use TORCH_LIBRARY() or TORCH_LIBRARY_IMPL() instead of using these
constructors directly
|
Copyright © 2024. All rights reserved.