Searched refs:ThenWaitFor (Results 1 – 11 of 11) sorted by relevance
137 send_device_to_host_stream->ThenWaitFor(send_stream); in SetProtoFromGPU()210 send_device_to_device_stream->ThenWaitFor(send_stream); in DeviceToDeviceCopy()229 send_device_to_device_stream->ThenWaitFor(recv_stream); in DeviceToDeviceCopy()277 send_device_to_host_stream->ThenWaitFor(send_stream); in CopyGPUTensorToCPU()322 recv_host_to_device_stream->ThenWaitFor(recv_stream); in CopyCPUTensorToGPU()
561 if (idc->stream() != stream) stream->ThenWaitFor(idc->stream()); in ComputeHelper()
132 stream->ThenWaitFor(FindOrDie(thunk_to_finish_event, dependency).get()); in ExecuteThunks()155 main_stream->ThenWaitFor(&sub_streams); in ExecuteThunks()
45 stream->ThenWaitFor(&sub_streams); in GetCyclesTaken()
89 stream->ThenWaitFor(definition_event_.get()); in WaitForDefinitionEventOnStream()
158 host_to_device_stream_->ThenWaitFor(stream_.get()); in CopyCPUTensorToDevice()
333 gpu_device_context->stream()->ThenWaitFor( in PopulateOutputs()
53 substream->ThenWaitFor(stream); in TransferLiteralFromDevice()103 substream->ThenWaitFor(stream); in TransferLiteralToDevice()
187 Stream &ThenWaitFor(Stream *other, Params... more_streams) { in ThenWaitFor() function188 return ThenWaitFor(more_streams...).ThenWaitFor(other); in ThenWaitFor()200 Stream &ThenWaitFor(Stream *other);206 Stream &ThenWaitFor(P others) { in ThenWaitFor() function209 ThenWaitFor(stream.get()); in ThenWaitFor()218 Stream &ThenWaitFor(Event *event);
1844 Stream &Stream::ThenWaitFor(Stream *other) { in ThenWaitFor() function in stream_executor::Stream1858 Stream &Stream::ThenWaitFor(Event *event) { in ThenWaitFor() function in stream_executor::Stream
513 nccl_stream->stream->ThenWaitFor(p->tensor_stream); in RunCollective()