Home
last modified time | relevance | path

Searched refs:WaitForBufferDefinitionEventsOnStream (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtracked_device_buffer.h251 void WaitForBufferDefinitionEventsOnStream(const TrackedDeviceBuffer& buffer,
Dtracked_device_buffer.cc228 void WaitForBufferDefinitionEventsOnStream(const TrackedDeviceBuffer& buffer, in WaitForBufferDefinitionEventsOnStream() function
Dpjrt_stream_executor_client.cc1235 WaitForBufferDefinitionEventsOnStream(*device_buffer, stream); in ToLiteral()
1305 WaitForBufferDefinitionEventsOnStream(*src_device_buffer, transfer_stream); in CopyToDeviceHelper()