From d40d28cfa27bbde39d315f03326155dbbc708ac7 Mon Sep 17 00:00:00 2001
From: 0nullpointer <null@ente.io>
Date: Tue, 9 Jan 2024 16:18:00 +0530
Subject: [PATCH] avoid sqrt computation for text encoder

---
 .../semantic_search/frameworks/onnx/onnx_text_encoder.dart     | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/lib/services/semantic_search/frameworks/onnx/onnx_text_encoder.dart b/lib/services/semantic_search/frameworks/onnx/onnx_text_encoder.dart
index a27bdb53a..5c9927085 100644
--- a/lib/services/semantic_search/frameworks/onnx/onnx_text_encoder.dart
+++ b/lib/services/semantic_search/frameworks/onnx/onnx_text_encoder.dart
@@ -51,8 +51,9 @@ class OnnxTextEncoder {
       textNormalization += embedding[i] * embedding[i];
     }
 
+    final double sqrtTexteNormalization = sqrt(textNormalization);
     for (int i = 0; i < 512; i++) {
-      embedding[i] = embedding[i] / sqrt(textNormalization);
+      embedding[i] = embedding[i] / sqrtTexteNormalization;
     }
 
     return (embedding);