diff options
| -rw-r--r-- | src/core/NEON/kernels/NEReductionOperationKernel.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/core/NEON/kernels/NEReductionOperationKernel.cpp b/src/core/NEON/kernels/NEReductionOperationKernel.cpp index e2dee67d01..9b8c971a35 100644 --- a/src/core/NEON/kernels/NEReductionOperationKernel.cpp +++ b/src/core/NEON/kernels/NEReductionOperationKernel.cpp @@ -1047,7 +1047,7 @@ struct RedOpYZW_quantized wrapper::vstore(reinterpret_cast<uint32_t *>(output.ptr()) + 8, vec_res_idx.val[2]); wrapper::vstore(reinterpret_cast<uint32_t *>(output.ptr()) + 12, vec_res_idx.val[3]); } - else if(op == ReductionOperation::ARG_IDX_MIN) + else if(op == ReductionOperation::MIN || op == ReductionOperation::MAX) { wrapper::vstore(reinterpret_cast<T *>(output.ptr()), vec_res_value); } |