blob: 829ed7fae136268f8f8c65e1e7df1aae93bb0556 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#pragma once
#include "distance_function.h"
#include "bound_distance_function.h"
#include <vespa/eval/eval/value_type.h>
#include <vespa/searchcommon/attribute/distance_metric.h>
namespace search::tensor {
/**
* API for binding the LHS of a distance calculation
* This allows keeping global state in the factory itself, and state
* for one particular vector in the distance function object.
*/
struct DistanceFunctionFactory {
DistanceFunctionFactory() = default;
virtual ~DistanceFunctionFactory() {}
virtual BoundDistanceFunction::UP for_query_vector(const vespalib::eval::TypedCells& lhs) = 0;
virtual BoundDistanceFunction::UP for_insertion_vector(const vespalib::eval::TypedCells& lhs) = 0;
using UP = std::unique_ptr<DistanceFunctionFactory>;
};
/**
* Create a distance function factory customized for the given metric
* variant and (attribute) cell type.
**/
DistanceFunctionFactory::UP
make_distance_function_factory(search::attribute::DistanceMetric variant,
vespalib::eval::CellType cell_type);
}
|
