diff options
author | Tor Egge <Tor.Egge@online.no> | 2022-10-07 20:38:42 +0200 |
---|---|---|
committer | Tor Egge <Tor.Egge@online.no> | 2022-10-07 20:38:42 +0200 |
commit | 57ff64a20d652c54a868c3fea977cc8aced82750 (patch) | |
tree | 4f0df998c4a788a5bb113d574f8bb9db5e17c1fc /searchlib/src/tests | |
parent | 36bc9cc2d6d01746c42e3041d4e2f000ebb4a7dd (diff) |
Share code for loading and saving tensor attribute between
SerializedFastValueAttribute and DirectTensorAttribute.
Diffstat (limited to 'searchlib/src/tests')
-rw-r--r-- | searchlib/src/tests/tensor/dense_tensor_store/dense_tensor_store_test.cpp | 6 |
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); } |