tensorplay.compiler.substitute_in_graph

tensorplay.compiler.substitute_in_graph(original_fn: Callable[[_P], _R], *, can_constant_fold_through: bool = False, skip_signature_check: bool = False) → Callable[[Callable[[_P], _R]], Callable[[_P], _R]][source]

Register a graph-time implementation for a callable.