RegistrationHandleRAII |
Dispatcher.addRegistrationListener(OpRegistrationListener listener)
Add a listener that gets called whenever a new op is registered or an existing
op is deregistered.
|
RegistrationHandleRAII |
RegistrationHandleRAII.put(RegistrationHandleRAII rhs) |
RegistrationHandleRAII |
Dispatcher.registerDef(FunctionSchema schema,
BytePointer debug) |
RegistrationHandleRAII |
Dispatcher.registerDef(FunctionSchema schema,
BytePointer debug,
int[] tags) |
RegistrationHandleRAII |
Dispatcher.registerDef(FunctionSchema schema,
BytePointer debug,
IntBuffer tags) |
RegistrationHandleRAII |
Dispatcher.registerDef(FunctionSchema schema,
BytePointer debug,
IntPointer tags)
Register a new operator schema.
|
RegistrationHandleRAII |
Dispatcher.registerDef(FunctionSchema schema,
String debug) |
RegistrationHandleRAII |
Dispatcher.registerDef(FunctionSchema schema,
String debug,
int[] tags) |
RegistrationHandleRAII |
Dispatcher.registerDef(FunctionSchema schema,
String debug,
IntBuffer tags) |
RegistrationHandleRAII |
Dispatcher.registerDef(FunctionSchema schema,
String debug,
IntPointer tags) |
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) |
RegistrationHandleRAII |
Dispatcher.registerLibrary(BytePointer ns,
BytePointer debug)
Use to register whenever we had a TORCH_LIBRARY declaration in the frontend
API.
|
RegistrationHandleRAII |
Dispatcher.registerLibrary(String ns,
String debug) |
RegistrationHandleRAII |
Dispatcher.registerName(OperatorName op_name)
Register a new operator by name.
|