1 // 2 // Copyright © 2017 Arm Ltd and Contributors. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 #pragma once 6 7 #include <Layer.hpp> 8 9 namespace armnn 10 { 11 12 /// This layer converts data type Float 32 to Float 16. 13 class ConvertFp32ToFp16Layer : public Layer 14 { 15 public: 16 /// Makes a workload for the ConvertFp32ToFp16 type. 17 /// @param [in] factory The workload factory which will create the workload. 18 /// @return A pointer to the created workload, or nullptr if not created. 19 virtual std::unique_ptr<IWorkload> CreateWorkload(const IWorkloadFactory& factory) const override; 20 21 /// Creates a dynamically-allocated copy of this layer. 22 /// @param [in] graph The graph into which this layer is being cloned. 23 ConvertFp32ToFp16Layer* Clone(Graph& graph) const override; 24 25 /// Check if the input tensor shape(s) 26 /// will lead to a valid configuration of @ref ConvertFp32ToFp16Layer. 27 /// @param [in] shapeInferenceMethod Indicates if output shape shall be overwritten or just validated. 28 void ValidateTensorShapesFromInputs() override; 29 30 void ExecuteStrategy(IStrategy& strategy) const override; 31 32 protected: 33 /// Constructor to create a ConvertFp32ToFp16Layer. 34 /// @param [in] name Optional name for the layer. 35 ConvertFp32ToFp16Layer(const char* name); 36 37 /// Default destructor 38 ~ConvertFp32ToFp16Layer() = default; 39 }; 40 41 } // namespace 42