1*89c4ff92SAndroid Build Coastguard Worker // 2*89c4ff92SAndroid Build Coastguard Worker // Copyright © 2023 Arm Ltd and Contributors. All rights reserved. 3*89c4ff92SAndroid Build Coastguard Worker // SPDX-License-Identifier: MIT 4*89c4ff92SAndroid Build Coastguard Worker // 5*89c4ff92SAndroid Build Coastguard Worker 6*89c4ff92SAndroid Build Coastguard Worker #pragma once 7*89c4ff92SAndroid Build Coastguard Worker 8*89c4ff92SAndroid Build Coastguard Worker #include "LayerWithParameters.hpp" 9*89c4ff92SAndroid Build Coastguard Worker 10*89c4ff92SAndroid Build Coastguard Worker namespace armnn 11*89c4ff92SAndroid Build Coastguard Worker { 12*89c4ff92SAndroid Build Coastguard Worker 13*89c4ff92SAndroid Build Coastguard Worker /// This layer represents a elementwiseBinary operation. 14*89c4ff92SAndroid Build Coastguard Worker class ElementwiseBinaryLayer : public LayerWithParameters<ElementwiseBinaryDescriptor> 15*89c4ff92SAndroid Build Coastguard Worker { 16*89c4ff92SAndroid Build Coastguard Worker public: 17*89c4ff92SAndroid Build Coastguard Worker /// Makes a workload for the elementwiseBinary type 18*89c4ff92SAndroid Build Coastguard Worker /// @param [in] graph The graph where this layer can be found 19*89c4ff92SAndroid Build Coastguard Worker /// @param [in] factory The workload factory which will create the workload 20*89c4ff92SAndroid Build Coastguard Worker /// @return A pointer to the created workload, or nullptr if not created 21*89c4ff92SAndroid Build Coastguard Worker virtual std::unique_ptr<IWorkload> CreateWorkload(const IWorkloadFactory& factory) const override; 22*89c4ff92SAndroid Build Coastguard Worker 23*89c4ff92SAndroid Build Coastguard Worker /// Creates a dynamically-allocated copy of this layer 24*89c4ff92SAndroid Build Coastguard Worker /// @param [in] graph The graph into which this layer is being cloned 25*89c4ff92SAndroid Build Coastguard Worker ElementwiseBinaryLayer* Clone(Graph& graph) const override; 26*89c4ff92SAndroid Build Coastguard Worker 27*89c4ff92SAndroid Build Coastguard Worker /// Returns inputShapes by default. 28*89c4ff92SAndroid Build Coastguard Worker /// @param [in] inputShapes The input shapes layer has. 29*89c4ff92SAndroid Build Coastguard Worker /// @return A vector to the inferred output shape. 30*89c4ff92SAndroid Build Coastguard Worker std::vector<TensorShape> InferOutputShapes(const std::vector<TensorShape>& inputShapes) const override; 31*89c4ff92SAndroid Build Coastguard Worker 32*89c4ff92SAndroid Build Coastguard Worker /// Check if the input tensor shape(s) will lead to a valid configuration 33*89c4ff92SAndroid Build Coastguard Worker /// of @ref ElementwiseBinaryLayer 34*89c4ff92SAndroid Build Coastguard Worker void ValidateTensorShapesFromInputs() override; 35*89c4ff92SAndroid Build Coastguard Worker 36*89c4ff92SAndroid Build Coastguard Worker void ExecuteStrategy(IStrategy& strategy) const override; 37*89c4ff92SAndroid Build Coastguard Worker 38*89c4ff92SAndroid Build Coastguard Worker protected: 39*89c4ff92SAndroid Build Coastguard Worker /// Constructor to create a ElementwiseBinaryLayer 40*89c4ff92SAndroid Build Coastguard Worker /// @param [in] param ElementwiseBinaryDescriptor to configure the ElementwiseBinaryLayer 41*89c4ff92SAndroid Build Coastguard Worker /// @param [in] name Optional name for the layer 42*89c4ff92SAndroid Build Coastguard Worker ElementwiseBinaryLayer(const ElementwiseBinaryDescriptor& param, const char* name); 43*89c4ff92SAndroid Build Coastguard Worker 44*89c4ff92SAndroid Build Coastguard Worker /// Default destructor 45*89c4ff92SAndroid Build Coastguard Worker ~ElementwiseBinaryLayer() = default; 46*89c4ff92SAndroid Build Coastguard Worker }; 47*89c4ff92SAndroid Build Coastguard Worker 48*89c4ff92SAndroid Build Coastguard Worker } // namespace armnn 49