chore: reverse timesteps and embeddings to support descending order o…

…f the timesteps and embeddings
mlcommons · Dec 29, 2024 · b035768 · b035768
1 parent 0835157
commit b035768
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/mobile_back_tflite/cpp/backend_tflite/embedding_utils.cc b/mobile_back_tflite/cpp/backend_tflite/embedding_utils.cc
@@ -25,6 +25,10 @@ bool TsEmbeddingParser::parse_pickle(const std::string& filename) {
               EMBEDDING_DIM * sizeof(float));
   }
 
+  // Reverse both timesteps and embeddings before storing
+    std::reverse(timesteps.begin(), timesteps.end());
+    std::reverse(embeddings.begin(), embeddings.end());
+
   // Store in maps
   timesteps_[num_timesteps] = std::move(timesteps);
   embeddings_[num_timesteps] = std::move(embeddings);