nccl_manager.cc | 95 Participant(const Tensor* in_t, Tensor* out_t, EventMgr* event_mgr, 101 event_mgr(event_mgr), 107 DCHECK(event_mgr != nullptr); 119 EventMgr* const event_mgr; member in struct:tensorflow::NcclManager::Participant 302 int gpu_device_id, EventMgr* event_mgr, 307 new Participant(in_t, out_t, event_mgr, tensor_stream, executor, 316 EventMgr* event_mgr, perftools::gputools::Stream* tensor_stream, 319 new Participant(in_t, nullptr /* out_t */, event_mgr, tensor_stream, 329 EventMgr* event_mgr, perftools::gputools::Stream* tensor_stream [all...] |