aboutsummaryrefslogtreecommitdiffstats
path: root/eval/src/vespa/eval/instruction/l2_distance.h
blob: 5d50b4ee2836f0860cefe9171a0a0f80cc63e073 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
// Copyright Vespa.ai. 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