diff options
| author | Matthew Bentham <matthew.bentham@arm.com> | 2021-07-07 09:08:48 +0100 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2021-07-16 15:43:54 +0000 |
| commit | 18bf43d9caeb7f482e0beaef45202b6638d48a8c (patch) | |
| tree | 81edfa19c9a44f100e2fe814a097ae95c7ca36e4 /src/backends/reference/workloads/FullyConnected.hpp | |
| parent | 94df84d1e4964f00fbeaa29a720ef1822ba033d6 (diff) | |
| download | armnn-18bf43d9caeb7f482e0beaef45202b6638d48a8c.tar.gz | |
Fix binding of reference to null pointer in RefFullyConnectedWorkload
As the bias decoder is optional, pass it as a pointer to the FullyConnected
function instead of by reference.
Signed-off-by: Matthew Bentham <matthew.bentham@arm.com>
Change-Id: I0bd2e601015e7fa1ae21b50f6c3dd7eac4176554
Diffstat (limited to 'src/backends/reference/workloads/FullyConnected.hpp')
| -rw-r--r-- | src/backends/reference/workloads/FullyConnected.hpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/backends/reference/workloads/FullyConnected.hpp b/src/backends/reference/workloads/FullyConnected.hpp index e33060631b..d7881f0b66 100644 --- a/src/backends/reference/workloads/FullyConnected.hpp +++ b/src/backends/reference/workloads/FullyConnected.hpp @@ -21,7 +21,7 @@ void FullyConnected(const TensorShape& rInputShape, Encoder<float>& rOutputEncoder, const TensorShape& rWeightsShape, Decoder<float>& rWeightDecoder, - Decoder<float>& rBiasDecoder, + Decoder<float>* rBiasDecoder, bool biasEnabled, unsigned int K, bool transposeWeights); |