Searched refs:_create_stateful_graph_module (Results 1 – 2 of 2) sorted by relevance
115 from torch.export._unlift import _create_stateful_graph_module182 module = _create_stateful_graph_module(
256 def _create_stateful_graph_module( function357 unlift_gm = _create_stateful_graph_module(