From 164b83e251da2333be25ad1768cef205a0023262 Mon Sep 17 00:00:00 2001 From: Håvard Pettersen Date: Wed, 29 Nov 2017 14:50:13 +0000 Subject: optimize for NN bias addition --- eval/src/vespa/eval/tensor/dense/dense_tensor_view.cpp | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 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 eba2452c41d..30c9f17348e 100644 --- a/eval/src/vespa/eval/tensor/dense/dense_tensor_view.cpp +++ b/eval/src/vespa/eval/tensor/dense/dense_tensor_view.cpp @@ -241,16 +241,16 @@ DenseTensorView::accept(TensorVisitor &visitor) const Tensor::UP DenseTensorView::join(join_fun_t function, const Tensor &arg) const { - if (function == eval::operation::Mul::f) { - if (fast_type() == arg.type()) { - return joinDenseTensors(*this, arg, "match", - [](double lhsValue, double rhsValue) - { return (lhsValue * rhsValue); }); - } else { - return dense::apply(*this, arg, - [](double lhsValue, double rhsValue) - { return lhsValue * rhsValue; }); + if (fast_type() == arg.type()) { + if (function == eval::operation::Mul::f) { + return joinDenseTensors(*this, arg, "mul", + [](double a, double b) { return (a * b); }); } + if (function == eval::operation::Add::f) { + return joinDenseTensors(*this, arg, "add", + [](double a, double b) { return (a + b); }); + } + return joinDenseTensors(*this, arg, "join", function); } return dense::apply(*this, arg, function); } -- cgit v1.2.3