Home
last modified time | relevance | path

Searched refs:tensor_stream (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager.h68 Participant(se::StreamExecutor* executor, se::Stream* tensor_stream, in Participant()
72 tensor_stream(tensor_stream), in Participant()
85 DCHECK(tensor_stream != nullptr); in Participant()
97 se::Stream* const tensor_stream; member
Dnccl_manager.cc635 nccl_stream->stream->ThenWaitFor(p->tensor_stream); in RunCollective()