Searched defs:nccl_communicator (Results 1 – 4 of 4) sorted by relevance
30 std::unique_ptr<NcclCommunicatorInterface> nccl_communicator) in CollectiveExecutorMgr()
32 std::unique_ptr<NcclCommunicatorInterface> nccl_communicator, in RpcCollectiveExecutorMgr()
170 CollectiveExecutor* col_exec, NcclCommunicatorInterface* nccl_communicator, in CollectiveContext()
366 NcclCommunicatorInterface* nccl_communicator; // Not owned member