Home
last modified time | relevance | path

Searched refs:are_deterministic_algorithms_enabled (Results 1 – 24 of 24) sorted by relevance

/aosp_15_r20/external/pytorch/test/quantization/core/experimental/
H A Dtest_float8.py267 with DeterministicGuard(torch.are_deterministic_algorithms_enabled()):
/aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/
H A Dautograd_cache.py174 self.deterministic_algorithms = torch.are_deterministic_algorithms_enabled()
H A Djit_compile_runtime_wrappers.py332 fw_metadata.deterministic = torch.are_deterministic_algorithms_enabled()
H A Druntime_wrappers.py1658 global_deterministic = torch.are_deterministic_algorithms_enabled()
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dtorch.py472 @register(torch.are_deterministic_algorithms_enabled)
475 return ConstantVariable.create(torch.are_deterministic_algorithms_enabled())
H A Dctx_manager.py654 initial_values=[torch.are_deterministic_algorithms_enabled()],
/aosp_15_r20/external/pytorch/docs/source/
H A Dtorch.compiler_cudagraph_trees.rst255 …e when `torch.are_deterministic_algorithms_enabled() <https://pytorch.org/docs/stable/generated/to…
H A Dbackends.rst124 See also :func:`torch.are_deterministic_algorithms_enabled` and
H A Dtorch.rst701 are_deterministic_algorithms_enabled
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dutils.py739 if torch.are_deterministic_algorithms_enabled():
1849 or torch.are_deterministic_algorithms_enabled()
H A Dlowering.py3174 deterministic = torch.are_deterministic_algorithms_enabled()
3212 if torch.are_deterministic_algorithms_enabled():
6164 torch.are_deterministic_algorithms_enabled()
H A Dcodecache.py667 torch.are_deterministic_algorithms_enabled(),
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_guard_manager.py85 _orig = torch.are_deterministic_algorithms_enabled()
H A Dtest_misc.py7342 torch.are_deterministic_algorithms_enabled(),
7407 prior = torch.are_deterministic_algorithms_enabled()
7417 value = torch.are_deterministic_algorithms_enabled()
7431 assert torch.are_deterministic_algorithms_enabled() is False
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dconvert_frame.py202 prior_deterministic = torch.are_deterministic_algorithms_enabled()
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.py4571 if torch.are_deterministic_algorithms_enabled() and (
5086 if torch.are_deterministic_algorithms_enabled() and (
/aosp_15_r20/external/pytorch/torch/
H A D__init__.py1342 def are_deterministic_algorithms_enabled() -> builtins.bool: function
H A Doverrides.py294 torch.are_deterministic_algorithms_enabled,
/aosp_15_r20/external/pytorch/torch/_prims_common/
H A D__init__.py1953 if torch.are_deterministic_algorithms_enabled():
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_utils.py1887 self.deterministic_restore = torch.are_deterministic_algorithms_enabled()
1975 torch.are_deterministic_algorithms_enabled(),
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dtriton.py2533 "are_deterministic_algorithms_enabled": torch.are_deterministic_algorithms_enabled(),
/aosp_15_r20/external/pytorch/test/
H A Dtest_transformers.py61 previous_mode: bool = torch.are_deterministic_algorithms_enabled()
H A Dtest_torch.py9595 self.assertEqual(deterministic, torch.are_deterministic_algorithms_enabled())
9614 self.assertEqual(deterministic, torch.are_deterministic_algorithms_enabled())
H A Dtest_sparse.py1433 with DeterministicGuard(torch.are_deterministic_algorithms_enabled()):