HomeSort by relevance Sort by last modified time
    Searched refs:ThenWaitFor (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_util.cc 139 send_device_to_host_stream->ThenWaitFor(send_stream);
214 send_device_to_device_stream->ThenWaitFor(send_stream);
233 send_device_to_device_stream->ThenWaitFor(recv_stream);
281 send_device_to_host_stream->ThenWaitFor(send_stream);
326 recv_host_to_device_stream->ThenWaitFor(recv_stream);
gpu_device.cc 479 LOG(INFO) << " stream[" << stream_id << "].ThenWaitFor(stream["
484 if (idc->stream() != stream) stream->ThenWaitFor(idc->stream());
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_executable.cc 176 stream->ThenWaitFor(FindOrDie(thunk_to_finish_event, dependency).get());
180 stream->ThenWaitFor(FindOrDie(thunk_to_finish_event,
218 main_stream->ThenWaitFor(&sub_streams);
  /external/tensorflow/tensorflow/stream_executor/
stream.h 175 Stream &ThenWaitFor(Stream *other, Params... more_streams) {
176 return ThenWaitFor(more_streams...).ThenWaitFor(other);
188 // N.B. Base recursion case for the variadic ThenWaitFor.
189 Stream &ThenWaitFor(Stream *other);
195 Stream &ThenWaitFor(P others) {
198 ThenWaitFor(stream.get());
207 Stream &ThenWaitFor(Event *event);
    [all...]
stream.cc     [all...]
  /external/tensorflow/tensorflow/contrib/nccl/kernels/
nccl_manager.cc 416 nccl_stream->stream->ThenWaitFor(p->tensor_stream);

Completed in 319 milliseconds