src/armnn/layers/ConvertFp16ToFp32Layer.hpp

   1 //
   2 // Copyright © 2017 Arm Ltd. All rights reserved.
   3 // See LICENSE file in the project root for full license information.
   4 //
   5
   6 #pragma once
   7
   8 #include <Layer.hpp>
   9
  10 namespace armnn
  11 {
  12
  13 class ConvertFp16ToFp32Layer : public Layer
  14 {
  15 public:
  16     virtual std::unique_ptr<IWorkload> CreateWorkload(const Graph& graph,
  17                                                       const IWorkloadFactory& factory) const override;
  18
  19     ConvertFp16ToFp32Layer* Clone(Graph& graph) const override;
  20
  21     void ValidateTensorShapesFromInputs() override;
  22
  23 protected:
  24     ConvertFp16ToFp32Layer(const char* name);
  25     ~ConvertFp16ToFp32Layer() = default;
  26 };
  27
  28 } // namespace