Searched refs:WaitForEventOnStream (Results 1 – 3 of 3) sorted by relevance
51 void BufferSequencingEvent::WaitForEventOnStream(se::Stream* stream) { in WaitForEventOnStream() function in xla::BufferSequencingEvent233 event->WaitForEventOnStream(stream); in WaitForBufferDefinitionEventsOnStream()
72 void WaitForEventOnStream(se::Stream* stream);
1075 stream_and_event.event->WaitForEventOnStream(stream.get()); in Release()1100 stream_and_event.event->WaitForEventOnStream(block_stream.get()); in Release()1457 event->WaitForEventOnStream(stream.get()); in BlockHostUntilReady()1672 event->WaitForEventOnStream(device_state->compute_stream()); in MakeExecutionInputsAndWaitForEvents()