Searched defs:ReleaseDeviceMemoryOwnership (Results 1 – 3 of 3) sorted by relevance
352 StatusOr<std::unique_ptr<ExternalReference>> ReleaseDeviceMemoryOwnership( in ReleaseDeviceMemoryOwnership() function
713 TfrtCpuBuffer::ReleaseDeviceMemoryOwnership( in ReleaseDeviceMemoryOwnership() function in xla::TfrtCpuBuffer
671 PjRtStreamExecutorBuffer::ReleaseDeviceMemoryOwnership( in ReleaseDeviceMemoryOwnership() function in xla::PjRtStreamExecutorBuffer