From 8186b38bb64eed8838419153f8f137a92c76e61a Mon Sep 17 00:00:00 2001 From: 0nullpointer Date: Tue, 9 Jan 2024 16:22:28 +0530 Subject: [PATCH] Avoid multiple sqrt computation for text encoder --- .../semantic_search/frameworks/onnx/onnx_text_encoder.dart | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) 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..664b925e7 100644 --- a/lib/services/semantic_search/frameworks/onnx/onnx_text_encoder.dart +++ b/lib/services/semantic_search/frameworks/onnx/onnx_text_encoder.dart @@ -50,9 +50,10 @@ class OnnxTextEncoder { for (int i = 0; i < 512; i++) { textNormalization += embedding[i] * embedding[i]; } - + + final double sqrtTextNormalization = sqrt(textNormalization); for (int i = 0; i < 512; i++) { - embedding[i] = embedding[i] / sqrt(textNormalization); + embedding[i] = embedding[i] / sqrtTextNormalization; } return (embedding);