diff options
author | Håvard Pettersen <havardpe@oath.com> | 2017-11-29 14:50:13 +0000 |
---|---|---|
committer | Håvard Pettersen <havardpe@oath.com> | 2017-11-29 14:50:13 +0000 |
commit | 164b83e251da2333be25ad1768cef205a0023262 (patch) | |
tree | 9e99f988c95edcd53a41730031c94cc82d273d61 | |
parent | 166ce7c53a96f733d5e1b909bc8b465a2f8d70ec (diff) |
optimize for NN bias addition
-rw-r--r-- | eval/src/vespa/eval/tensor/dense/dense_tensor_view.cpp | 18 |
1 files 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); } |