Share code for loading and saving tensor attribute between

SerializedFastValueAttribute and DirectTensorAttribute.

1 files changed, 3 insertions, 3 deletions

diff --git a/searchlib/src/tests/tensor/dense_tensor_store/dense_tensor_store_test.cpp b/searchlib/src/tests/tensor/dense_tensor_store/dense_tensor_store_test.cpp
index 149662cd266..26ef57aab65 100644
--- a/searchlib/src/tests/tensor/dense_tensor_store/dense_tensor_store_test.cpp
+++ b/searchlib/src/tests/tensor/dense_tensor_store/dense_tensor_store_test.cpp
@@ -32,15 +32,15 @@ struct Fixture
     {}
     void assertSetAndGetTensor(const TensorSpec &tensorSpec) {
         Value::UP expTensor = makeTensor(tensorSpec);
-        EntryRef ref = store.setTensor(*expTensor);
-        Value::UP actTensor = store.getTensor(ref);
+        EntryRef ref = store.store_tensor(*expTensor);
+        Value::UP actTensor = store.get_tensor(ref);
         EXPECT_EQUAL(*expTensor, *actTensor);
         assertTensorView(ref, *expTensor);
     }
     void assertEmptyTensor(const TensorSpec &tensorSpec) {
         Value::UP expTensor = makeTensor(tensorSpec);
         EntryRef ref;
-        Value::UP actTensor = store.getTensor(ref);
+        Value::UP actTensor = store.get_tensor(ref);
         EXPECT_TRUE(actTensor.get() == nullptr);
         assertTensorView(ref, *expTensor);
     }