Home
last modified time | relevance | path

Searched refs:is_fx_tracing (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/nested/
H A D__init__.py389 from torch.fx._symbolic_trace import is_fx_tracing
390 if is_fx_tracing():
/aosp_15_r20/external/pytorch/torch/export/
H A Dunflatten.py24 from torch.fx._symbolic_trace import is_fx_tracing
109 if not is_fx_tracing() and (
448 if is_fx_tracing():
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Deval_frame.py429 is_fx_tracing = torch.fx._symbolic_trace.is_fx_tracing
433 if is_fx_tracing():
/aosp_15_r20/external/pytorch/torch/fx/
H A D_symbolic_trace.py49 def is_fx_tracing(): function
105 if not is_fx_tracing():
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_functional_api.py404 self.assertFalse(torch.fx._symbolic_trace.is_fx_tracing())
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dtorch.py140 torch.fx._symbolic_trace.is_fx_tracing: False,
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_functions.py462 if torch.fx._symbolic_trace.is_fx_tracing():