Searched refs:OnDeviceSizeInBytes (Results 1 – 6 of 6) sorted by relevance
71 int64 OnDeviceSizeInBytes() { return buffer_->OnDeviceSizeInBytes(); } in OnDeviceSizeInBytes() function
275 buffer->buffer()->OnDeviceSizeInBytes(), in HeapProfile()
307 .def("on_device_size_in_bytes", &PyBuffer::OnDeviceSizeInBytes) in PYBIND11_MODULE()
283 virtual int64 OnDeviceSizeInBytes() const = 0;
479 int64 OnDeviceSizeInBytes() const override;
1017 int64 PjRtStreamExecutorBuffer::OnDeviceSizeInBytes() const { in OnDeviceSizeInBytes() function in xla::PjRtStreamExecutorBuffer