1 // 2 // Copyright © 2020 Arm Ltd and Contributors. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 #pragma once 6 7 #include "LayerWithParameters.hpp" 8 9 namespace armnn 10 { 11 12 /// This layer represents a fill operation. 13 class FillLayer : public LayerWithParameters<FillDescriptor> 14 { 15 public: 16 /// Makes a workload for the Fill layer. 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 FillLayer* Clone(Graph& graph) const override; 24 25 /// Check if the input tensor shape(s) 26 /// will lead to a valid configuration of @ref FillLayer. 27 /// @param [in] shapeInferenceMethod Indicates if output shape shall be overwritten or just validate. 28 void ValidateTensorShapesFromInputs() override; 29 30 void ExecuteStrategy(IStrategy& strategy) const override; 31 32 protected: 33 /// Constructor to create a FillLayer. 34 /// @param [in] descriptor to configure the fill operation. 35 /// @param [in] name Optional name for the layer. 36 FillLayer(const FillDescriptor& descriptor, const char* name); 37 38 /// Default destructor 39 ~FillLayer() = default; 40 }; 41 42 } // namespace 43