Searched refs:nccl_communicator (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_executor_mgr.cc | 30 std::unique_ptr<NcclCommunicatorInterface> nccl_communicator) in CollectiveExecutorMgr() argument
36 nccl_communicator_(std::move(nccl_communicator)), in CollectiveExecutorMgr()
|
| D | collective_executor_mgr.h | 33 std::unique_ptr<NcclCommunicatorInterface> nccl_communicator);
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.cc | 170 CollectiveExecutor* col_exec, NcclCommunicatorInterface* nccl_communicator, in CollectiveContext() argument
175 nccl_communicator(nccl_communicator), in CollectiveContext()
|
| D | collective.h | 366 NcclCommunicatorInterface* nccl_communicator; // Not owned member
380 NcclCommunicatorInterface* nccl_communicator,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_broadcaster.cc | 27 col_ctx_->nccl_communicator->Enqueue(col_ctx_, std::move(done)); in Run()
|
| D | collective_nccl_gatherer.cc | 27 col_ctx_->nccl_communicator->Enqueue(col_ctx_, std::move(done)); in Run()
|
| D | collective_nccl_reducer.cc | 94 col_ctx_->nccl_communicator->Enqueue(col_ctx_, std::move(done_callback)); in Run()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | rpc_collective_executor_mgr.cc | 32 std::unique_ptr<NcclCommunicatorInterface> nccl_communicator, in RpcCollectiveExecutorMgr() argument
36 std::move(nccl_communicator)), in RpcCollectiveExecutorMgr()
|
| D | rpc_collective_executor_mgr.h | 41 std::unique_ptr<NcclCommunicatorInterface> nccl_communicator,
|