| Package | Description |
|---|---|
| org.bytedeco.pytorch | |
| org.bytedeco.pytorch.global |
| Modifier and Type | Method and Description |
|---|---|
OperatorName |
Library._resolve(BytePointer name)
Register an implementation for an operator.
|
OperatorName |
Library._resolve(String name) |
OperatorName |
OperatorNameOptional.get() |
OperatorName |
Dispatcher.getAllOpNames() |
OperatorName |
Dispatcher.getRegistrationsForDispatchKey(DispatchKeyOptional k)
Useful for inspecting global Dispatcher registration state.
|
OperatorName |
OperatorName.name(BytePointer setter) |
OperatorName |
FunctionSchema.operator_name() |
OperatorName |
OperatorHandle.operator_name() |
OperatorName |
OperatorName.overload_name(BytePointer setter) |
| Modifier and Type | Method and Description |
|---|---|
OperatorHandleOptional |
Dispatcher.findOp(OperatorName operator_name) |
OperatorHandleOptional |
Dispatcher.findSchema(OperatorName operator_name)
Looks for an operator schema with the given name and overload name
and returns it if it is registered WITH A SCHEMA.
|
OperatorNameOptional |
OperatorNameOptional.put(OperatorName value) |
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) |
RegistrationHandleRAII |
Dispatcher.registerName(OperatorName op_name)
Register a new operator by name.
|
void |
Dispatcher.waitForImpl(OperatorName op_name,
DispatchKeyOptional dispatch_key) |
| Constructor and Description |
|---|
OperatorNameOptional(OperatorName value) |
| Modifier and Type | Method and Description |
|---|---|
static OperatorName |
torch.parseName(BytePointer name) |
static OperatorName |
torch.parseName(String name) |
| Modifier and Type | Method and Description |
|---|---|
static boolean |
torch.equals(OperatorName lhs,
OperatorName rhs) |
static Operator |
torch.findOperatorFor(OperatorName full_name) |
static boolean |
torch.is_custom_op(OperatorName opName) |
static boolean |
torch.notEquals(OperatorName lhs,
OperatorName rhs) |
static Pointer |
torch.shiftLeft(Pointer arg0,
OperatorName arg1) |
static BytePointer |
torch.toString(OperatorName opName) |
Copyright © 2024. All rights reserved.