Reset file position back to begin before read

tensorflow · Sep 8, 2023 · 80c9a38 · 80c9a38
1 parent 2765581
commit 80c9a38
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/codegen/preprocessor/main.cc b/codegen/preprocessor/main.cc
@@ -34,6 +34,7 @@ std::unique_ptr<char[]> ReadModelFile(const char* model_file_name) {
 
   model_file.seekg(0, std::ios::end);
   size_t num_bytes = model_file.tellg();
+  model_file.seekg(0,std::ios::beg);
   std::unique_ptr<char[]> model_data(new char[num_bytes]);
   model_file.read(model_data.get(), num_bytes);