| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 135 send_device_to_host_stream->ThenWaitFor(send_stream); in SetProtoFromGPU()
209 send_device_to_device_stream->ThenWaitFor(send_stream); in DeviceToDeviceCopy()
228 send_device_to_device_stream->ThenWaitFor(recv_stream); in DeviceToDeviceCopy()
276 send_device_to_host_stream->ThenWaitFor(send_stream); in CopyGPUTensorToCPU()
322 recv_host_to_device_stream->ThenWaitFor(recv_stream); in CopyCPUTensorToGPU()
|
| D | gpu_device.cc | 264 group->compute->ThenWaitFor(group->nccl); in GetOrCreate()
265 group->nccl->ThenWaitFor(group->compute); in GetOrCreate()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | collective_permute_thunk.cc | 155 dest_stream->ThenWaitFor(src_stream).ThenMemcpyD2D(&dest, src, src.size()); in EnqueueCopy()
156 src_stream->ThenWaitFor(dest_stream); in EnqueueCopy()
|
| D | hlo_execution_profiler.cc | 44 stream->ThenWaitFor(&sub_streams); in GetCyclesTaken()
|
| D | gpu_executable.cc | 155 sub_streams.back()->ThenWaitFor(main_stream); in ExecuteThunks()
179 stream->ThenWaitFor(FindOrDie(thunk_to_finish_event, dependency).get()); in ExecuteThunks()
208 main_stream->ThenWaitFor(&sub_streams); in ExecuteThunks()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_tpu_device.cc | 222 dst_device_to_device_stream->ThenWaitFor(dst_compute_stream); in TpuDeviceToDeviceCopy()
227 dst_xla_context->host_to_device_stream()->ThenWaitFor( in TpuDeviceToDeviceCopy()
259 dst_device_to_device_stream->ThenWaitFor( in TpuDeviceToDeviceCopy()
|
| D | xla_tensor.cc | 86 stream->ThenWaitFor(definition_event_.get()); in WaitForDefinitionEventOnStream()
|
| D | xla_device_context.cc | 166 host_to_device_stream_->ThenWaitFor(stream_.get()); in CopyCPUTensorToDevice()
|
| D | xla_launch_util.cc | 375 gpu_device_context->stream()->ThenWaitFor( in SetOutputForConstant()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | local_device_state.h | 152 callback_stream_->ThenWaitFor(stream); in ThenRelease()
|
| D | tracked_device_buffer.cc | 67 stream->ThenWaitFor(event_.event()); in WaitForEventOnStream()
|
| D | pjrt_stream_executor_client.cc | 283 local_device->compute_stream()->ThenWaitFor(stream); in StallStreamOnError()
384 copy_stream->ThenWaitFor(local_device->compute_stream()); in AllocateDestinationBuffer()
428 tuple_table_stream->ThenWaitFor(local_device->compute_stream()); in AllocateDestinationBuffer()
806 local_device->callback_stream()->ThenWaitFor( in BufferFromHostBuffer()
1510 stream->ThenWaitFor(local_device->compute_stream()); in MakeTupleHelper()
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr_test.cc | 383 gpu_helper_->h2d_stream()->ThenWaitFor(gpu_helper_->compute_stream()); in DoAddChain()
394 gpu_helper_->compute_stream()->ThenWaitFor(gpu_helper_->h2d_stream()); in DoAddChain()
418 gpu_helper_->d2h_stream()->ThenWaitFor(gpu_helper_->compute_stream()); in DoAddChain()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | transfer_manager.cc | 56 substream->ThenWaitFor(stream); in TransferLiteralFromDevice()
106 substream->ThenWaitFor(stream); in TransferLiteralToDevice()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executable_interface.cc | 43 stream->ThenWaitFor(transfer_stream); in PopulateResultTupleBuffers()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.h | 201 Stream &ThenWaitFor(Stream *other, Params... more_streams) { in ThenWaitFor() function
202 return ThenWaitFor(more_streams...).ThenWaitFor(other); in ThenWaitFor()
214 Stream &ThenWaitFor(Stream *other);
220 Stream &ThenWaitFor(P others) { in ThenWaitFor() function
223 ThenWaitFor(stream.get()); in ThenWaitFor()
232 Stream &ThenWaitFor(Event *event);
|
| D | stream.cc | 1396 Stream &Stream::ThenWaitFor(Stream *other) { in ThenWaitFor() function in stream_executor::Stream
1410 Stream &Stream::ThenWaitFor(Event *event) { in ThenWaitFor() function in stream_executor::Stream
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 373 deregister_stream->ThenWaitFor(stream); in UnregisterCancellation()
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor_test.cc | 483 dependent.ThenWaitFor(&other); in TEST_F()
596 stream.ThenWaitFor(&event); in TEST_F()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_execute_op.cc | 696 stream->ThenWaitFor(transfer_stream_ptr.get()); in DoWork()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.cc | 635 nccl_stream->stream->ThenWaitFor(p->tensor_stream); in RunCollective()
|