diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2023-07-13 17:16:45 +0100 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2023-07-25 09:54:47 +0000 |
commit | 79a06a59bafadf736ca53c4240e87f9bbb657260 (patch) | |
tree | b94b2063766b40b79bed8cb81f571ea05cfa9efb /include/armnn/INetwork.hpp | |
parent | a638f101bcb51008932f922fe0a5cef28633bc66 (diff) | |
download | armnn-79a06a59bafadf736ca53c4240e87f9bbb657260.tar.gz |
IVGCVSW-7883 Front end and reference implementation for TILE
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Signed-off-by: Cian McGriskin <cian.mcgriskin@arm.com>
Change-Id: I0afb2403fee11c5c1e58ea65e2525e99594d8f2d
Diffstat (limited to 'include/armnn/INetwork.hpp')
-rw-r--r-- | include/armnn/INetwork.hpp | 27 |
1 files changed, 17 insertions, 10 deletions
diff --git a/include/armnn/INetwork.hpp b/include/armnn/INetwork.hpp index e20dd1c348..830e0bac66 100644 --- a/include/armnn/INetwork.hpp +++ b/include/armnn/INetwork.hpp @@ -740,7 +740,7 @@ public: /// @param name - Optional name for the layer. /// @return - Interface for configuring the layer. IConnectableLayer* AddGatherLayer(const GatherDescriptor& descriptor, - const char* name = nullptr); + const char* name = nullptr); /// Add GatherNd layer to the network. /// @param name - Optional name for the layer. @@ -764,23 +764,23 @@ public: /// @param name - Optional name for the layer. /// @return - Interface for configuring the layer. IConnectableLayer* AddTransposeConvolution2dLayer(const TransposeConvolution2dDescriptor& descriptor, - const ConstTensor& weights, - const Optional<ConstTensor>& biases, - const char* name = nullptr); + const ConstTensor& weights, + const Optional<ConstTensor>& biases, + const char* name = nullptr); /// Adds a transpose layer to the network. /// @param transposeDescriptor - TransposeDescriptor to configure the transpose. /// @param name - Optional name for the layer. /// @return - Interface for configuring the layer. IConnectableLayer* AddTransposeLayer(const TransposeDescriptor& transposeDescriptor, - const char* name = nullptr); + const char* name = nullptr); /// Adds a stack layer to the network. /// @param descriptor - Description of the stack layer. /// @param name - Optional name for the layer. /// @return - Interface for configuring the layer. IConnectableLayer* AddStackLayer(const StackDescriptor& descriptor, - const char* name = nullptr); + const char* name = nullptr); /// Add a stand-in layer for a type unknown to the Arm NN framework. /// Note: Due to the nature of this layer, no validation can be performed by the framework. @@ -789,14 +789,14 @@ public: /// @descriptor - Descriptor for the StandIn layer. /// @return - Interface for configuring the layer. IConnectableLayer* AddStandInLayer(const StandInDescriptor& descriptor, - const char* name = nullptr); + const char* name = nullptr); /// Add a QuantizedLstm layer to the network /// @param params - The weights and biases for the Quantized LSTM cell /// @param name - Optional name for the layer /// @return - Interface for configuring the layer. IConnectableLayer* AddQuantizedLstmLayer(const QuantizedLstmInputParams& params, - const char* name = nullptr); + const char* name = nullptr); /// Add a QLstm layer to the network /// @param descriptor - Parameters for the QLstm operation @@ -804,8 +804,8 @@ public: /// @param name - Optional name for the layer /// @return - Interface for configuring the layer. IConnectableLayer* AddQLstmLayer(const QLstmDescriptor& descriptor, - const LstmInputParams& params, - const char* name = nullptr); + const LstmInputParams& params, + const char* name = nullptr); /// Adds a Logical Binary layer to the network. /// @param descriptor - Description of the Logical Binary layer. @@ -842,6 +842,13 @@ public: /// @return - Interface for configuring the layer IConnectableLayer* AddReverseV2Layer(const char* name = nullptr); + /// Add a Tile layer to the network + /// @param descriptor - Parameters for the Tile operation + /// @param name - Optional name for the layer + /// @return - Interface for configuring the layer + IConnectableLayer* AddTileLayer(const TileDescriptor& descriptor, + const char* name = nullptr); + void ExecuteStrategy(IStrategy& strategy) const; protected: |