From 636ab40d3741e12eaad11d5b50e4b34bfbb258b5 Mon Sep 17 00:00:00 2001 From: Aron Virginas-Tar Date: Mon, 16 Sep 2019 14:27:45 +0100 Subject: IVGCVSW-3875 Add frontend for SLICE layer Signed-off-by: Aron Virginas-Tar Change-Id: Iebe675a0cee02db6f133d48ce58cbc1e233061db --- include/armnn/INetwork.hpp | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) (limited to 'include/armnn/INetwork.hpp') diff --git a/include/armnn/INetwork.hpp b/include/armnn/INetwork.hpp index cd1b7a6319..09026ad9e6 100644 --- a/include/armnn/INetwork.hpp +++ b/include/armnn/INetwork.hpp @@ -237,6 +237,12 @@ public: virtual IConnectableLayer* AddNormalizationLayer(const NormalizationDescriptor& normalizationDescriptor, const char* name = nullptr) = 0; + /// Adds a slice layer to the network. + /// @param sliceDescriptor - SliceDescriptor to configure the slice operation. + /// @param name - Optional name for the layer. + /// @return - Interface for configuring the layer. + virtual IConnectableLayer* AddSliceLayer(const SliceDescriptor& sliceDescriptor, const char* name = nullptr) = 0; + /// Adds a softmax layer to the network. /// If the data type is QAsymm8, then the output quantization parameters /// must have a scale of 1/256 and an offset of 0 @@ -253,8 +259,8 @@ public: /// the first output, second view to the second output, etc.... /// @param name - Optional name for the layer. /// @return - Interface for configuring the layer. - virtual IConnectableLayer* AddSplitterLayer(const ViewsDescriptor& splitterDescriptor - , const char* name = nullptr) = 0; + virtual IConnectableLayer* AddSplitterLayer(const ViewsDescriptor& splitterDescriptor, + const char* name = nullptr) = 0; /// Adds a merge layer to the network. /// @param name - Optional name for the layer. -- cgit v1.2.1