Searched refs:_use_lazy_graph_module (Results 1 – 4 of 4) sorted by relevance
14 _use_lazy_graph_module,27 cls.exit_stack.enter_context(_use_lazy_graph_module(True))
40 def _use_lazy_graph_module(should_use: bool): function
58 from ..fx._lazy_graph_module import _use_lazy_graph_module # type: ignore[attr-defined]434 @_use_lazy_graph_module(dynamo_config.use_lazy_graph_module)1258 @_use_lazy_graph_module(dynamo_config.use_lazy_graph_module)
40 from torch.fx._lazy_graph_module import _use_lazy_graph_module813 with _use_lazy_graph_module(config.use_lazy_graph_module), compile_context(