Operator Registration#

General Operator Customization Interface#

XLACustomOp

Creating a XLA custom call operator.

CPU Operator Customization with Numba#

CustomOpByNumba

Creating a XLA custom call operator with Numba JIT on CPU backend.

XLACustomOp

Creating a XLA custom call operator.

compile_cpu_signature_with_numba(c, func, ...)