Searched refs:ReleaseDeviceMemoryOwnership (Results 1 – 4 of 4) sorted by relevance
359 ReleaseDeviceMemoryOwnership(bool wait_for_operations_to_complete) = 0;
484 StatusOr<std::unique_ptr<ExternalReference>> ReleaseDeviceMemoryOwnership(
628 PjRtStreamExecutorBuffer::ReleaseDeviceMemoryOwnership( in ReleaseDeviceMemoryOwnership() function in xla::PjRtStreamExecutorBuffer
265 buffer->buffer()->ReleaseDeviceMemoryOwnership( in BufferToDLPackManagedTensor()