Searched defs:profiler_ctx (Results 1 – 1 of 1) sorted by relevance
1597 def _test_send_recv_nccl(self, profiler_ctx=None): argument1674 def _test_send_recv(self, profiler_ctx): argument1739 def _test_send_recv_any_source(self, profiler_ctx): argument1845 def _test_send_recv_with_tag(self, profiler_ctx): argument1905 def _test_isend(self, profiler_ctx): argument6881 def _test_ddp_profiling(self, profiler_ctx, profiler_ctx2=None): argument