diff options
Diffstat (limited to 'src/cpu/kernels/elementwise_unary/generic/neon/impl.cpp')
| -rw-r--r-- | src/cpu/kernels/elementwise_unary/generic/neon/impl.cpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/src/cpu/kernels/elementwise_unary/generic/neon/impl.cpp b/src/cpu/kernels/elementwise_unary/generic/neon/impl.cpp index 2b23e46f40..1b449340ba 100644 --- a/src/cpu/kernels/elementwise_unary/generic/neon/impl.cpp +++ b/src/cpu/kernels/elementwise_unary/generic/neon/impl.cpp @@ -183,6 +183,10 @@ void elementwise_op<int8_t>(const ITensor *in, ITensor *out, const Window &windo { tmp_f = (127 - qi_out.offset) * qi_out.scale; } + else + { + tmp_f = elementwise_op_scalar_imp<float>(op, tmp_f); + } } else { @@ -263,6 +267,10 @@ void elementwise_op<uint8_t>(const ITensor *in, ITensor *out, const Window &wind { tmp_f = (255 - qi_out.offset) * qi_out.scale; } + else + { + tmp_f = elementwise_op_scalar_imp<float>(op, tmp_f); + } } else { |