OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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