Searched defs:nccl_communicator (Results 1 – 6 of 6) sorted by relevance
33 std::unique_ptr<NcclCommunicatorInterface> nccl_communicator) in CollectiveExecutorMgr()99 std::unique_ptr<NcclCommunicatorInterface> nccl_communicator) { in CreateProdLocalCollectiveExecutorMgr()
48 NcclCommunicatorInterface* nccl_communicator, const string& task_name) in CollectiveParamResolverLocal()
32 std::unique_ptr<NcclCommunicatorInterface> nccl_communicator, in RpcCollectiveExecutorMgr()174 std::unique_ptr<NcclCommunicatorInterface> nccl_communicator, in CreateProdRpcCollectiveExecutorMgr()
92 NcclCommunicatorInterface* nccl_communicator, in CollectiveParamResolverDistributed()
156 CollectiveExecutor* col_exec, NcclCommunicatorInterface* nccl_communicator, in CollectiveContext()
393 NcclCommunicatorInterface* nccl_communicator; // Not owned member