diff options
| author | Finn Williams <Finn.Williams@arm.com> | 2020-09-29 19:54:00 +0100 |
|---|---|---|
| committer | TeresaARM <teresa.charlinreyes@arm.com> | 2020-10-01 07:57:12 +0000 |
| commit | ea8ce7040476da46e145705b0b08e9449144a3b2 (patch) | |
| tree | d6d09864e689a11e206ebfdd738d0191d06f6e91 /src/backends/reference/workloads/Pooling2d.cpp | |
| parent | 6d9f5c57fe80b3b3c08294ddd52062e107151a15 (diff) | |
| download | armnn-ea8ce7040476da46e145705b0b08e9449144a3b2.tar.gz | |
IVGCVSW-5325 Fix non-channel per axis quantization
Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: Ie0cf69b2cd76d6ecedab43d3d9ae267d23bbc052
Diffstat (limited to 'src/backends/reference/workloads/Pooling2d.cpp')
| -rw-r--r-- | src/backends/reference/workloads/Pooling2d.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/backends/reference/workloads/Pooling2d.cpp b/src/backends/reference/workloads/Pooling2d.cpp index 2bc3b4f213..be6ff387f3 100644 --- a/src/backends/reference/workloads/Pooling2d.cpp +++ b/src/backends/reference/workloads/Pooling2d.cpp @@ -180,7 +180,7 @@ void Pooling2d(Decoder<float>& rInputDecoder, throw armnn::InvalidArgumentException("Unsupported padding type"); } - const std::vector<float> decodedInputVec = rInputDecoder.DecodeTensor(inputInfo.GetNumElements()); + const std::vector<float> decodedInputVec = rInputDecoder.DecodeTensor(inputInfo.GetShape()); for (int n = 0; n < batchSize; n++) { |