Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_autograd.py12689 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 …]