tensorplay.compiler.disallow_in_graph

tensorplay.compiler.disallow_in_graph(fn: Callable[[...], Any]) → Callable[[...], Any][source]

Declare that a callable must not be absorbed into a captured graph.

The tracer is meant to treat calls to fn as region boundaries: the surrounding pieces are captured as separate graphs and fn itself runs eagerly between them. Under eager execution the marker is inert metadata, so the callable is returned unchanged.