tensorplay.library.register_autocast

tensorplay.library.register_autocast(op: str | CustomOpDef, device_type: str, cast_inputs: Any, /, *, lib: Any = None) → None[source]