aboutsummaryrefslogtreecommitdiff
path: root/src/core/NEON/kernels/arm_gemm/quantized.hpp
diff options
context:
space:
mode:
authorGeorgios Pinitas <georgios.pinitas@arm.com>2020-07-01 12:35:30 +0100
committerGeorgios Pinitas <georgios.pinitas@arm.com>2020-07-02 18:12:47 +0000
commitaf56d520b486079e5f773a530582a0a710f7f376 (patch)
treefa9c1041211da6a4a98170f56f19eb53b3763c40 /src/core/NEON/kernels/arm_gemm/quantized.hpp
parentd355653e3e967667a7d380c9c2ae2016207a6046 (diff)
downloadComputeLibrary-af56d520b486079e5f773a530582a0a710f7f376.tar.gz
COMPMID-3324: Fix per-channel quantization on N blocking
Direct the column to start from in the quantized code Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com> Change-Id: I8231e0b541c6b1b76becf349a1d6ddf973ade9e2 Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3488 Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/NEON/kernels/arm_gemm/quantized.hpp')
-rw-r--r--src/core/NEON/kernels/arm_gemm/quantized.hpp2
1 files changed, 1 insertions, 1 deletions
diff --git a/src/core/NEON/kernels/arm_gemm/quantized.hpp b/src/core/NEON/kernels/arm_gemm/quantized.hpp
index a91a888ad9..b0e0c3b580 100644
--- a/src/core/NEON/kernels/arm_gemm/quantized.hpp
+++ b/src/core/NEON/kernels/arm_gemm/quantized.hpp
@@ -28,7 +28,7 @@ namespace arm_gemm {
template<typename Tin, typename Tout>
void requantize_block_32(const Requantize32 &qp, unsigned int width, unsigned int height,
const Tin *input, unsigned int in_stride, Tout *output, unsigned int out_stride,
- const int32_t *row_bias, const int32_t *col_bias);
+ const int32_t *row_bias, const int32_t *col_bias, unsigned int start_col);
template<typename T>
void compute_row_sums(const Requantize32 &qp, unsigned int width, unsigned int height,