Website Blog TensorPlay documentation

Automatic differentiation package - tensorplay.autograd Automatic Mixed Precision package - tensorplay.amp Distributed communication package - tensorplay.distributed Index Multiprocessing package - tensorplay.multiprocessing Python Module Index Quantization Search tensorplay tensorplay._stax tensorplay.audio tensorplay.cuda tensorplay.fft tensorplay.futures tensorplay.hub tensorplay.library tensorplay.linalg tensorplay.nn tensorplay.nn.functional tensorplay.nn.init tensorplay.optim tensorplay.random tensorplay.sparse tensorplay.special tensorplay.stax tensorplay.utils.checkpoint tensorplay.utils.data tensorplay.vision

Library tensorplay.library.custom_op tensorplay.library.define tensorplay.library.get_kernel tensorplay.library.impl tensorplay.library.impl_abstract tensorplay.library.infer_schema tensorplay.library.opcheck tensorplay.library.register_autocast tensorplay.library.register_autograd tensorplay.library.register_fake tensorplay.library.register_kernel tensorplay.library.register_vmap tensorplay.library.triton_op tensorplay.library.wrap_triton

API symbolslibrary

latest (dev)

latest (dev)2026-09-08 v1.0.02026-09-06

Copy

tensorplay.library.impl

tensorplay.library.impl(qualname: str, types: Any, func: Callable[[...], Any] | None = None, /, *, lib: Any = None) → Callable[[...], Any] | Callable[[Callable[[...], Any]], Callable[[...], Any]][source]: types accepts concrete devices ("CPU"/"CUDA") or composite spellings (CompositeExplicitAutograd → the device-agnostic slot).

Help improve this page

Found an error, an unclear step, or a missing example?

Was this page helpful?

Suggest an edit Report an issue

tensorplay.library.get_kernel

dispatch_key accepts "cpu" / "cuda" / "default" and the composite spellings. Raises LookupError when nothing usable is registered (a disabled concrete kernel counts as absent, matching CustomOpDef.set_kernel_enabled() vi

tensorplay.library.impl_abstract

Next Page

On this page