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/FullyConnected.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/FullyConnected.cpp')
-rw-r--r-- | src/backends/reference/workloads/FullyConnected.cpp | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/src/backends/reference/workloads/FullyConnected.cpp b/src/backends/reference/workloads/FullyConnected.cpp index 61c8e88bce..9ec9ea6c6c 100644 --- a/src/backends/reference/workloads/FullyConnected.cpp +++ b/src/backends/reference/workloads/FullyConnected.cpp @@ -24,10 +24,11 @@ void FullyConnected(const TensorShape& rInputShape, // Perform FullyConnected implementation unsigned int outputSize = rOutputShape[1]; - const std::vector<float> decodedInputs = rInputDecoder.DecodeTensor(rInputShape.GetNumElements()); - const std::vector<float> decodedWeights = rWeightDecoder.DecodeTensor(rWeightsShape.GetNumElements()); - const std::vector<float> decodedBiases = biasEnabled ? - rBiasDecoder.DecodeTensor(outputSize) : std::vector<float>(); + const std::vector<float> decodedInputs = rInputDecoder.DecodeTensor(rInputShape); + const std::vector<float> decodedWeights = rWeightDecoder.DecodeTensor(rWeightsShape); + + const TensorShape biasShape{outputSize}; + const std::vector<float> decodedBiases = biasEnabled ? rBiasDecoder.DecodeTensor(biasShape) : std::vector<float>(); for (unsigned int n = 0; n < rInputShape[0]; n++) |