From 5edc8816118fcddb2681379db04c978041ce8b46 Mon Sep 17 00:00:00 2001 From: Aron Virginas-Tar Date: Tue, 5 Nov 2019 18:00:21 +0000 Subject: IVGCVSW-3837 Add support for per-axis quantization to reference Convolution2d workload Signed-off-by: Aron Virginas-Tar Change-Id: I0ac08ba4864d48e6f64c4ac645dad8ea850be112 --- src/backends/reference/workloads/ConvImpl.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'src/backends/reference/workloads/ConvImpl.cpp') diff --git a/src/backends/reference/workloads/ConvImpl.cpp b/src/backends/reference/workloads/ConvImpl.cpp index 92e3b2d7dd..0c13e3ba0d 100644 --- a/src/backends/reference/workloads/ConvImpl.cpp +++ b/src/backends/reference/workloads/ConvImpl.cpp @@ -165,7 +165,7 @@ void Convolve(const TensorShape& rInputShape, } } - rFilterDecoder[filterIndex]; + rFilterDecoder.SetIndex(filterIndex, cOutput); float filterValue = rFilterDecoder.Get(); unsigned int yInput = yOutput * yStride + yFilter * yDilation; @@ -211,7 +211,7 @@ void Convolve(const TensorShape& rInputShape, if (biasEnabled) { - (*pBiasDecoder)[cOutput]; + (*pBiasDecoder).SetIndex(cOutput, cOutput); sum += pBiasDecoder->Get(); } @@ -225,4 +225,4 @@ void Convolve(const TensorShape& rInputShape, } } -} //namespace armnn +} // namespace armnn -- cgit v1.2.1