tensorplay.compiler.allow_in_graph

tensorplay.compiler.allow_in_graph(fn: Any) → Any[source]

Capture a callable as one graph operation without entering its body.