Home
last modified time | relevance | path

Searched refs:is_dynamo_compiling (Results 1 – 11 of 11) sorted by relevance

/external/pytorch/docs/source/
Dtorch.compiler_api.rst25 is_dynamo_compiling
Dtorch.compiler_fine_grain_apis.rst30 …"``torch.compiler.is_dynamo_compiling``", "Indicates whether a graph is traced via TorchDynamo. It…
/external/pytorch/torch/compiler/
D__init__.py314 def is_dynamo_compiling() -> bool: function
/external/pytorch/torch/_higher_order_ops/
Dstrict_mode.py21 if torch.compiler.is_dynamo_compiling():
Dwhile_loop.py120 if torch.compiler.is_dynamo_compiling():
Dcond.py129 if torch.compiler.is_dynamo_compiling():
/external/pytorch/torch/_dynamo/variables/
Dtorch.py145 torch.compiler.is_dynamo_compiling: True,
380 torch.compiler.is_dynamo_compiling,
/external/pytorch/torch/export/
Dunflatten.py110 torch.compiler.is_dynamo_compiling() or not self._run_with_interpeter
504 if torch.compiler.is_dynamo_compiling() and not self._run_with_interpreter:
/external/pytorch/torch/nn/attention/
Dflex_attention.py1014 if torch.compiler.is_dynamo_compiling():
/external/pytorch/torch/distributed/
D_functional_collectives.py29 from torch.compiler import is_dynamo_compiling as is_torchdynamo_compiling
/external/pytorch/test/dynamo/
Dtest_misc.py149 assert torch.compiler.is_dynamo_compiling() or UserDefineSetAttr.setup
153 assert torch.compiler.is_dynamo_compiling() or UserDefineSetAttr.setup
6223 if torch.compiler.is_dynamo_compiling():