summaryrefslogtreecommitdiffstats
path: root/eval/src/tests
diff options
context:
space:
mode:
authorHåvard Pettersen <havardpe@oath.com>2018-02-02 12:50:13 +0000
committerHåvard Pettersen <havardpe@oath.com>2018-02-02 12:50:13 +0000
commitfa03629bf9039951fe5ad92149cab295efac0620 (patch)
tree21caa62f4d1bbb551cd7f049763d337b4d33f2a3 /eval/src/tests
parentb6e4d2af8c058edc1487c3ca49262cac5502e043 (diff)
co-locate optimization with implementation
Diffstat (limited to 'eval/src/tests')
-rw-r--r--eval/src/tests/tensor/dense_tensor_function_optimizer/dense_tensor_function_optimizer_test.cpp5
1 files changed, 2 insertions, 3 deletions
diff --git a/eval/src/tests/tensor/dense_tensor_function_optimizer/dense_tensor_function_optimizer_test.cpp b/eval/src/tests/tensor/dense_tensor_function_optimizer/dense_tensor_function_optimizer_test.cpp
index c09b218508e..269a1e265c5 100644
--- a/eval/src/tests/tensor/dense_tensor_function_optimizer/dense_tensor_function_optimizer_test.cpp
+++ b/eval/src/tests/tensor/dense_tensor_function_optimizer/dense_tensor_function_optimizer_test.cpp
@@ -3,7 +3,6 @@
#include <vespa/vespalib/testkit/test_kit.h>
#include <vespa/eval/tensor/dense/dense_dot_product_function.h>
#include <vespa/eval/tensor/dense/dense_xw_product_function.h>
-#include <vespa/eval/tensor/dense/dense_tensor_function_optimizer.h>
#include <vespa/eval/eval/operation.h>
using namespace vespalib::eval;
@@ -23,7 +22,7 @@ optimizeDotProduct(const vespalib::string &lhsType,
inject(ValueType::from_spec(rhsType), 3, stash),
Mul::f, stash),
Aggr::SUM, {}, stash);
- return DenseTensorFunctionOptimizer::optimize(reduceNode, stash);
+ return DenseDotProductFunction::optimize(reduceNode, stash);
}
void assertParam(const TensorFunction &node, size_t expect_idx) {
@@ -66,7 +65,7 @@ optimizeXWProduct(const vespalib::string &lhsType,
inject(ValueType::from_spec(rhsType), 3, stash),
Mul::f, stash),
Aggr::SUM, {dim}, stash);
- return DenseTensorFunctionOptimizer::optimize(reduceNode, stash);
+ return DenseXWProductFunction::optimize(reduceNode, stash);
}
void