Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/
H A Ddist_autograd_test.py320 self._verify_graph_for_rpc_call_exec(next(iter(send_functions.values())))
411 self._verify_graph_for_rpc_call_exec(next(iter(send_functions.values())))
480 self._verify_graph_for_rpc_call_exec(list(send_functions.values())[1])
954 def _verify_graph_for_rpc_call_exec(self, send_function): member in CommonDistAutogradTest