diff options
| author | Francis Murtagh <francis.murtagh@arm.com> | 2019-05-27 12:14:10 +0100 |
|---|---|---|
| committer | Francis Murtagh <francis.murtagh@arm.com> | 2019-05-27 12:14:10 +0100 |
| commit | 43aec5886449c1b024b740fd6f4500e827bde221 (patch) | |
| tree | c12a128dcc6895a0663a4e4dd27c4110e492c6dd /src/backends/reference/workloads/FullyConnected.hpp | |
| parent | 7f2c35a82ec11be50b3478bd15207320bbf3bd57 (diff) | |
| download | armnn-43aec5886449c1b024b740fd6f4500e827bde221.tar.gz | |
IVGCVSW-3134 Refactor FullyConnected workloads into single workload
* Refactor FullyConnected workloads into single workload.
* Refactor FullyConnected ref implementation to use Encoders
and Decoders to support all DataTypes.
* Deleted RefFullyConnectedFloat32Workload and
RefFullyConnected2dUint8Workload.
Change-Id: Iad30fb0287ab7491e1297997e7d61f1d00785541
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Diffstat (limited to 'src/backends/reference/workloads/FullyConnected.hpp')
| -rw-r--r-- | src/backends/reference/workloads/FullyConnected.hpp | 20 |
1 files changed, 13 insertions, 7 deletions
diff --git a/src/backends/reference/workloads/FullyConnected.hpp b/src/backends/reference/workloads/FullyConnected.hpp index 623259f8f8..78fa055086 100644 --- a/src/backends/reference/workloads/FullyConnected.hpp +++ b/src/backends/reference/workloads/FullyConnected.hpp @@ -5,18 +5,24 @@ #pragma once +#include "BaseIterator.hpp" +#include "Decoders.hpp" +#include "Encoders.hpp" #include <armnn/Tensor.hpp> +#include <backendsCommon/WorkloadData.hpp> namespace armnn { /// Performs a matrix multiplication and optionally adds a bias. -void FullyConnected(const float* inputData, - float* outputData, - const TensorInfo& inputTensorInfo, - const TensorInfo& outputTensorInfo, - const float* weightData, - const float* biasData, - bool transposeWeights); +void FullyConnected(const TensorShape& rInputShape, + Decoder<float>& rInputDecoder, + const TensorShape& rOutputShape, + Encoder<float>& rOutputEncoder, + Decoder<float>& rWeightDecoder, + Decoder<float>& rBiasDecoder, + bool biasEnabled, + unsigned int K, + bool transposeWeights); } //namespace armnn |