blob: 03c976bd6b3d2b7cc2d4ad761c20c3a45f0ea9f2 (
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 "default_nearest_neighbor_index_factory.h"
#include "dense_tensor_store.h"
#include "tensor_attribute.h"
#include <memory>
namespace search::tensor {
/**
* Attribute vector class used to store dense tensors for all
* documents in memory.
*/
class DenseTensorAttribute : public TensorAttribute {
private:
DenseTensorStore _denseTensorStore;
public:
DenseTensorAttribute(vespalib::stringref baseFileName, const Config& cfg,
const NearestNeighborIndexFactory& index_factory = DefaultNearestNeighborIndexFactory());
~DenseTensorAttribute() override;
// Implements AttributeVector and ITensorAttribute
vespalib::eval::TypedCells extract_cells_ref(DocId docId) const override;
bool supports_extract_cells_ref() const override { return true; }
// Implements DocVectorAccess
vespalib::eval::TypedCells get_vector(uint32_t docid, uint32_t subspace) const override;
VectorBundle get_vectors(uint32_t docid) const override;
};
}
|