diff options
author | Finn Williams <Finn.Williams@arm.com> | 2020-09-17 15:58:31 +0100 |
---|---|---|
committer | finn.williams <finn.williams@arm.com> | 2020-09-28 09:01:58 +0000 |
commit | b9dcfe63b87f024c6f8c5f4b68447de04119dc19 (patch) | |
tree | 0c58376c59190ecbc8df0dd2abedbf85983d5256 /src/backends/reference/workloads/FullyConnected.hpp | |
parent | be727becad9fe048480ab53a0281b46594f95ca7 (diff) | |
download | armnn-b9dcfe63b87f024c6f8c5f4b68447de04119dc19.tar.gz |
IVGCVSW-5325 Speed up the reference backend
Change-Id: Id8bd0a0418be31d975b944b54bbacb25051ffb2e
Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Diffstat (limited to 'src/backends/reference/workloads/FullyConnected.hpp')
-rw-r--r-- | src/backends/reference/workloads/FullyConnected.hpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/FullyConnected.hpp b/src/backends/reference/workloads/FullyConnected.hpp index 6f9559db24..e33060631b 100644 --- a/src/backends/reference/workloads/FullyConnected.hpp +++ b/src/backends/reference/workloads/FullyConnected.hpp @@ -19,6 +19,7 @@ void FullyConnected(const TensorShape& rInputShape, Decoder<float>& rInputDecoder, const TensorShape& rOutputShape, Encoder<float>& rOutputEncoder, + const TensorShape& rWeightsShape, Decoder<float>& rWeightDecoder, Decoder<float>& rBiasDecoder, bool biasEnabled, |