Inline Mul and Add in reduce

1 files changed, 6 insertions, 0 deletions

diff --git a/eval/src/vespa/eval/tensor/dense/dense_tensor_view.cpp b/eval/src/vespa/eval/tensor/dense/dense_tensor_view.cpp
index 74c8981168d..749dc3dea7d 100644
--- a/eval/src/vespa/eval/tensor/dense/dense_tensor_view.cpp
+++ b/eval/src/vespa/eval/tensor/dense/dense_tensor_view.cpp
@@ -251,6 +251,12 @@ DenseTensorView::join(join_fun_t function, const Tensor &arg) const
         }
         return joinDenseTensors(*this, arg, "join", function);
     }
+    if (function == eval::operation::Mul::f) {
+        return dense::apply(*this, arg, [](double a, double b) { return (a * b); });
+    }
+    if (function == eval::operation::Add::f) {
+        return dense::apply(*this, arg, [](double a, double b) { return (a + b); });
+    }
     return dense::apply(*this, arg, function);
 }