diff options
Diffstat (limited to 'src/core/NEON/kernels/arm_gemm/quantized.cpp')
| -rw-r--r-- | src/core/NEON/kernels/arm_gemm/quantized.cpp | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/core/NEON/kernels/arm_gemm/quantized.cpp b/src/core/NEON/kernels/arm_gemm/quantized.cpp index 6da9f4be0e..dbc64fe038 100644 --- a/src/core/NEON/kernels/arm_gemm/quantized.cpp +++ b/src/core/NEON/kernels/arm_gemm/quantized.cpp @@ -745,6 +745,10 @@ template void requantize_block_32(const Requantize32 &qp, unsigned int width, un const uint32_t *input, unsigned int in_stride, uint8_t *output, unsigned int out_stride, const int32_t *row_bias, const int32_t *col_bias, unsigned int start_col); +template void requantize_block_32(const Requantize32 &qp, unsigned int width, unsigned int height, + const int32_t *input, unsigned int in_stride, uint8_t *output, unsigned int out_stride, + const int32_t *row_bias, const int32_t *col_bias, unsigned int start_col); + /* * Routine (and helpers) to compute row sums needed for offset correction. * @@ -966,7 +970,7 @@ void compute_row_sums(const Requantize32 &qp, unsigned int width, unsigned int h const unsigned int odds = width % 16; /* Generate a mask to use on the last iteration, if necessary. */ - uint64x2_t mask; + uint64x2_t mask = vdupq_n_u64(0); unsigned int mask_mode = 0; if (odds > 0 && odds <= 8) { |