Searched refs:wait_for_operations_to_complete (Results 1 – 3 of 3) sorted by relevance
485 bool wait_for_operations_to_complete) override;542 bool wait_for_operations_to_complete);
359 ReleaseDeviceMemoryOwnership(bool wait_for_operations_to_complete) = 0;
629 bool wait_for_operations_to_complete) { in ReleaseDeviceMemoryOwnership() argument636 Release(wait_for_operations_to_complete)); in ReleaseDeviceMemoryOwnership()1040 PjRtStreamExecutorBuffer::Release(bool wait_for_operations_to_complete) { in Release() argument1065 if (wait_for_operations_to_complete) { in Release()