Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/
H A Ddist_autograd_test.py334 def _test_graph_for_py_nested_call(self, exec_mode, sparse): member in CommonDistAutogradTest
1033 self._test_graph_for_py_nested_call(ExecMode.RPC_SYNC, True)
1037 self._test_graph_for_py_nested_call(ExecMode.REMOTE, True)
1327 self._test_graph_for_py_nested_call(ExecMode.RPC_SYNC, False)
1331 self._test_graph_for_py_nested_call(ExecMode.REMOTE, False)