2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // See LICENSE file in the project root for full license information.
8 #include "backends/Workload.hpp"
9 #include "backends/WorkloadData.hpp"
14 class RefBatchNormalizationUint8Workload : public Uint8Workload<BatchNormalizationQueueDescriptor>
17 explicit RefBatchNormalizationUint8Workload(const BatchNormalizationQueueDescriptor& descriptor,
18 const WorkloadInfo& info);
19 virtual void Execute() const override;
22 std::unique_ptr<ScopedCpuTensorHandle> m_Mean;
23 std::unique_ptr<ScopedCpuTensorHandle> m_Variance;
24 std::unique_ptr<ScopedCpuTensorHandle> m_Beta;
25 std::unique_ptr<ScopedCpuTensorHandle> m_Gamma;