src/backends/reference/workloads/RefBaseConstantWorkload.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 #include <armnn/Types.hpp>
  12
  13 namespace armnn
  14 {
  15
  16 // Base class template providing an implementation of the Constant layer common to all data types.
  17 template <armnn::DataType DataType>
  18 class RefBaseConstantWorkload : public TypedWorkload<ConstantQueueDescriptor, DataType>
  19 {
  20 public:
  21     RefBaseConstantWorkload(const ConstantQueueDescriptor& descriptor, const WorkloadInfo& info)
  22         : TypedWorkload<ConstantQueueDescriptor, DataType>(descriptor, info)
  23         , m_RanOnce(false)
  24     {
  25     }
  26
  27     virtual void Execute() const override;
  28
  29 private:
  30     mutable bool m_RanOnce;
  31 };
  32
  33 } //namespace armnn