Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtracked_device_buffer.cc214 void GetDeviceBufferEvents( in GetDeviceBufferEvents() function
231 GetDeviceBufferEvents(buffer, /*get_usage_events=*/false, &events); in WaitForBufferDefinitionEventsOnStream()
Dtracked_device_buffer.h246 void GetDeviceBufferEvents(const TrackedDeviceBuffer& buffer,
Dpjrt_stream_executor_client.cc1725 GetDeviceBufferEvents(*device_buffer, /*get_usage_events=*/must_donate, in EnqueueExecution()