Searched defs:GetOnDeviceSizeInBytes (Results 1 – 2 of 2) sorted by relevance
669 StatusOr<size_t> TfrtCpuBuffer::GetOnDeviceSizeInBytes() const { in GetOnDeviceSizeInBytes() function in xla::TfrtCpuBuffer
1363 StatusOr<size_t> PjRtStreamExecutorBuffer::GetOnDeviceSizeInBytes() const { in GetOnDeviceSizeInBytes() function in xla::PjRtStreamExecutorBuffer