Home
last modified time | relevance | path

Searched refs:set_multithreading_enabled (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DThreadLocalState.cpp34 void ThreadLocalState::set_multithreading_enabled(bool enabled) { in set_multithreading_enabled() function in at::ThreadLocalState
35 autograd_tls_.set_multithreading_enabled(enabled); in set_multithreading_enabled()
H A DThreadLocalState.h36 void set_multithreading_enabled(bool enabled);
/aosp_15_r20/external/pytorch/c10/core/
H A DAutogradState.h36 void set_multithreading_enabled(bool multithreading_enabled) { in set_multithreading_enabled() function
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Dmulti_threaded_pg.py413 self._ctx = torch.autograd.set_multithreading_enabled(False)
537 _ctx_manager = torch.autograd.set_multithreading_enabled(False)
/aosp_15_r20/external/pytorch/torch/autograd/
H A Dgrad_mode.py293 class set_multithreading_enabled(_DecoratorContextManager): class
H A D__init__.py29 set_multithreading_enabled,
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dinit.cpp918 static PyObject* set_multithreading_enabled( in set_multithreading_enabled() function
940 c10::AutogradState::get_tls_state().set_multithreading_enabled( in set_multithreading_enabled()
1359 castPyCFunctionWithKeywords(set_multithreading_enabled),
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dcompiled_autograd.py507 with torch.autograd.set_multithreading_enabled(False):
/aosp_15_r20/external/pytorch/docs/source/
H A Dautograd.rst309 grad_mode.set_multithreading_enabled
/aosp_15_r20/external/pytorch/torch/_functorch/
H A Daot_autograd.py586 with torch.autograd.set_multithreading_enabled(
/aosp_15_r20/external/pytorch/test/
H A Dtest_autograd.py4435 with torch.autograd.set_multithreading_enabled(False):
4452 with torch.autograd.set_multithreading_enabled(False):
4464 with torch.autograd.set_multithreading_enabled(False):
4477 with torch.autograd.set_multithreading_enabled(False):
4495 with torch.autograd.set_multithreading_enabled(False):
4506 with torch.autograd.set_multithreading_enabled(False):
4525 with torch.autograd.set_multithreading_enabled(False):
13075 with torch.autograd.set_multithreading_enabled(False):
13079 with torch.autograd.set_multithreading_enabled(True):
13083 with torch.autograd.set_multithreading_enabled(False):
[all …]
H A Dtest_ops.py2615 … with warnings.catch_warnings(), context(), torch.autograd.set_multithreading_enabled(
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_compiled_autograd.py90 with torch.autograd.set_multithreading_enabled(False):