diff options
Diffstat (limited to 'searchlib/src/vespa/searchlib/tensor/tensor_buffer_type_mapper.h')
-rw-r--r-- | searchlib/src/vespa/searchlib/tensor/tensor_buffer_type_mapper.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/searchlib/src/vespa/searchlib/tensor/tensor_buffer_type_mapper.h b/searchlib/src/vespa/searchlib/tensor/tensor_buffer_type_mapper.h index 950076c9924..ad2116a429c 100644 --- a/searchlib/src/vespa/searchlib/tensor/tensor_buffer_type_mapper.h +++ b/searchlib/src/vespa/searchlib/tensor/tensor_buffer_type_mapper.h @@ -25,12 +25,13 @@ public: using LargeBufferType = LargeSubspacesBufferType; TensorBufferTypeMapper(); - TensorBufferTypeMapper(uint32_t max_small_subspaces_type_id, TensorBufferOperations* ops); + TensorBufferTypeMapper(uint32_t max_small_subspaces_type_id, double grow_factor, TensorBufferOperations* ops); ~TensorBufferTypeMapper(); uint32_t get_type_id(size_t array_size) const; size_t get_array_size(uint32_t type_id) const; TensorBufferOperations& get_tensor_buffer_operations() const noexcept { return *_ops; } + uint32_t get_max_small_array_type_id(uint32_t max_small_array_type_id) const noexcept; }; } |