tensorplay.library.register_kernel

tensorplay.library.register_kernel(op: str | CustomOpDef, device_types: Any = None, func: Callable[[...], Any] | None = None, /, *, lib: Any = None) → Callable[[Callable[[...], Any]], Callable[[...], Any]][source]

Accepts a CustomOpDef or a qualified operator name. device_types=None or an empty iterable means the device-agnostic slot; composite spellings (Composite…) map there too. Usable directly or as a decorator.