summaryrefslogtreecommitdiffstats
path: root/eval/src/vespa/eval/instruction/l2_distance.h
diff options
context:
space:
mode:
Diffstat (limited to 'eval/src/vespa/eval/instruction/l2_distance.h')
-rw-r--r--eval/src/vespa/eval/instruction/l2_distance.h21
1 files changed, 21 insertions, 0 deletions
diff --git a/eval/src/vespa/eval/instruction/l2_distance.h b/eval/src/vespa/eval/instruction/l2_distance.h
new file mode 100644
index 00000000000..95b11b6c229
--- /dev/null
+++ b/eval/src/vespa/eval/instruction/l2_distance.h
@@ -0,0 +1,21 @@
+// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+
+#pragma once
+
+#include <vespa/eval/eval/tensor_function.h>
+
+namespace vespalib::eval {
+
+/**
+ * Tensor function for a squared euclidean distance producing a scalar result.
+ **/
+class L2Distance : public tensor_function::Op2
+{
+public:
+ L2Distance(const TensorFunction &lhs_in, const TensorFunction &rhs_in);
+ InterpretedFunction::Instruction compile_self(const ValueBuilderFactory &factory, Stash &stash) const override;
+ bool result_is_mutable() const override { return true; }
+ static const TensorFunction &optimize(const TensorFunction &expr, Stash &stash);
+};
+
+} // namespace