src/backends/reference/workloads/RefBatchNormalizationUint8Workload.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 RefBatchNormalizationUint8Workload : public Uint8Workload<BatchNormalizationQueueDescriptor>
  15 {
  16 public:
  17     explicit RefBatchNormalizationUint8Workload(const BatchNormalizationQueueDescriptor& descriptor,
  18                                           const WorkloadInfo& info);
  19     virtual void Execute() const override;
  20
  21 private:
  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;
  26 };
  27
  28 } //namespace armnn