IVGCVSW-3837 Add support for per-axis quantization to reference Convolution2d workload
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: I0ac08ba4864d48e6f64c4ac645dad8ea850be112
diff --git a/src/backends/reference/workloads/ConvImpl.cpp b/src/backends/reference/workloads/ConvImpl.cpp
index 92e3b2d..0c13e3b 100644
--- a/src/backends/reference/workloads/ConvImpl.cpp
+++ b/src/backends/reference/workloads/ConvImpl.cpp
@@ -165,7 +165,7 @@
}
}
- rFilterDecoder[filterIndex];
+ rFilterDecoder.SetIndex(filterIndex, cOutput);
float filterValue = rFilterDecoder.Get();
unsigned int yInput = yOutput * yStride + yFilter * yDilation;
@@ -211,7 +211,7 @@
if (biasEnabled)
{
- (*pBiasDecoder)[cOutput];
+ (*pBiasDecoder).SetIndex(cOutput, cOutput);
sum += pBiasDecoder->Get();
}
@@ -225,4 +225,4 @@
}
}
-} //namespace armnn
+} // namespace armnn