Searched defs:OnDeviceSizeInBytes (Results 1 – 2 of 2) sorted by relevance
71 int64 OnDeviceSizeInBytes() { return buffer_->OnDeviceSizeInBytes(); } in OnDeviceSizeInBytes() function
1017 int64 PjRtStreamExecutorBuffer::OnDeviceSizeInBytes() const { in OnDeviceSizeInBytes() function in xla::PjRtStreamExecutorBuffer