Searched refs:_run_py_multithread_fn (Results 1 – 1 of 1) sorted by relevance
/aosp_15_r20/external/pytorch/test/ |
H A D | test_autograd.py | 12689 def _run_py_multithread_fn( member in TestMultithreadAutograd 12728 self._run_py_multithread_fn(fn) 12739 self._run_py_multithread_fn(train_fn) 12749 self._run_py_multithread_fn(train_fn_backward, (x,)) 12765 self._run_py_multithread_fn(train_fn_grad, (x,)) 12797 self._run_py_multithread_fn(backward_retain_graph, (out, t2, t3), num_threads=5) 12833 self._run_py_multithread_fn(backward_retain_graph, (out, t2, t3), num_threads=5) 12870 self._run_py_multithread_fn(backward_retain_graph, (out, t2, t3), num_threads=5) 12905 self._run_py_multithread_fn(backward_retain_graph, (out, t2, t3), num_threads=5) 12966 self._run_py_multithread_fn( [all …]
|