IVGCVSW-1946: Remove armnn/src from the include paths
[platform/upstream/armnn.git] / src / backends / reference / workloads / RefFullyConnectedUint8Workload.hpp
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5
6 #pragma once
7
8 #include <backendsCommon/Workload.hpp>
9 #include <backendsCommon/WorkloadData.hpp>
10
11 namespace armnn
12 {
13
14 class RefFullyConnectedUint8Workload : public Uint8Workload<FullyConnectedQueueDescriptor>
15 {
16 public:
17     explicit RefFullyConnectedUint8Workload(const FullyConnectedQueueDescriptor& descriptor,
18                                              const WorkloadInfo& info);
19     virtual void Execute() const override;
20
21 private:
22     std::unique_ptr<ScopedCpuTensorHandle> m_Weight;
23     std::unique_ptr<ScopedCpuTensorHandle> m_Bias;
24 };
25
26 } //namespace armnn