From 824a95793cc3c37009bed907e15d7b8ff7cccd90 Mon Sep 17 00:00:00 2001
From: Rhett Stucki <rhettstucki@google.com>
Date: Fri, 6 Jun 2025 00:47:29 -0700
Subject: [PATCH] Fix Image::WriteBinary() writing values to a file one at a
 time.

PiperOrigin-RevId: 767955187
---
 paligemma/image.cc | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/paligemma/image.cc b/paligemma/image.cc
index f5bf4f2..20ecad8 100644
--- a/paligemma/image.cc
+++ b/paligemma/image.cc
@@ -201,9 +201,8 @@ bool Image::WriteBinary(const std::string& filename) const {
     std::cerr << "Failed to open " << filename << "\n";
     return false;
   }
-  for (size_t i = 0; i < data_.size(); ++i) {
-    file.write(reinterpret_cast<const char*>(&data_[i]), sizeof(float));
-  }
+  file.write(reinterpret_cast<const char*>(data_.data()),
+             data_.size() * sizeof(data_[0]));
   file.close();
   return true;
 }