Home
last modified time | relevance | path

Searched refs:use_deterministic_algorithms (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/pytorch/docs/source/notes/
H A Drandomness.rst94 :meth:`torch.use_deterministic_algorithms` lets you configure PyTorch to use
99 Please check the documentation for :meth:`torch.use_deterministic_algorithms()`
109 >>> torch.use_deterministic_algorithms(True)
114 'torch.use_deterministic_algorithms(True)'. ...
121 >>> torch.use_deterministic_algorithms(True)
137 :code:`torch.use_deterministic_algorithms(True)` or
139 controls only this behavior, unlike :meth:`torch.use_deterministic_algorithms`
156 :code:`torch.use_deterministic_algorithms(True)` is set. This will to prevent
/aosp_15_r20/external/pytorch/docs/source/
H A Ddeterministic.rst9 a known value when :meth:`torch.use_deterministic_algorithms()` is set to
H A Dbackends.rst125 :func:`torch.use_deterministic_algorithms`.
H A Dtorch.rst700 use_deterministic_algorithms
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Dnumeric_utils.py33 torch.use_deterministic_algorithms(True)
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_guard_manager.py87 torch.use_deterministic_algorithms(not _orig)
99 torch.use_deterministic_algorithms(_orig)
H A Dtest_misc.py7348 torch.use_deterministic_algorithms(state[1])
7423 torch.use_deterministic_algorithms(False, warn_only=False)
7427 torch.use_deterministic_algorithms(True, warn_only=True)
7440 torch.use_deterministic_algorithms(prior, warn_only=prior_warn_only)
9613 torch.use_deterministic_algorithms(forward_deterministic)
9617 torch.use_deterministic_algorithms(backward_deterministic)
/aosp_15_r20/external/pytorch/test/quantization/core/experimental/
H A Dtest_float8.py269 torch.use_deterministic_algorithms(use_deterministic)
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dtorch.py464 @register(torch.use_deterministic_algorithms)
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dconvert_frame.py225 torch.use_deterministic_algorithms(
/aosp_15_r20/external/pytorch/torch/
H A D__init__.py1200 def use_deterministic_algorithms( function
H A Doverrides.py295 torch.use_deterministic_algorithms,
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_utils.py1890 torch.use_deterministic_algorithms(
1896 torch.use_deterministic_algorithms(
/aosp_15_r20/external/pytorch/test/
H A Dtest_transformers.py64 torch.use_deterministic_algorithms(mode, warn_only=warn_only)
67 torch.use_deterministic_algorithms(previous_mode, warn_only=previous_warn_only)
H A Dtest_sparse.py1434 torch.use_deterministic_algorithms(False)
1436 torch.use_deterministic_algorithms(True)
H A Dtest_torch.py9594 torch.use_deterministic_algorithms(deterministic, warn_only=warn_only)
9624 torch.use_deterministic_algorithms(1)
9629 torch.use_deterministic_algorithms(False, warn_only=1)
H A Dtest_mps.py11350 torch.use_deterministic_algorithms(deterministic)
11358 torch.use_deterministic_algorithms(False)
/aosp_15_r20/external/pytorch/benchmarks/dynamo/
H A Dcommon.py3765 torch.use_deterministic_algorithms(True)