Home
last modified time | relevance | path

Searched defs:wait_for_operations_to_complete (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_c_api_client.h353 bool wait_for_operations_to_complete) override { in ReleaseDeviceMemoryOwnership()
H A Dtfrt_cpu_pjrt_client.cc714 bool wait_for_operations_to_complete) { in ReleaseDeviceMemoryOwnership()
787 bool wait_for_operations_to_complete) { in Release()
H A Dpjrt_stream_executor_client.cc672 bool wait_for_operations_to_complete) { in ReleaseDeviceMemoryOwnership()
1148 PjRtStreamExecutorBuffer::Release(bool wait_for_operations_to_complete) { in Release()