Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtfrt_cpu_pjrt_client.cc720 bool wait_for_operations_to_complete) { in ReleaseDeviceMemoryOwnership()
762 bool wait_for_operations_to_complete) { in Release()
Dpjrt_stream_executor_client.cc662 bool wait_for_operations_to_complete) { in ReleaseDeviceMemoryOwnership()
1149 PjRtStreamExecutorBuffer::Release(bool wait_for_operations_to_complete) { in Release()