diff options
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++) { |