diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-09-16 14:27:45 +0100 |
---|---|---|
committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-09-17 09:03:43 +0000 |
commit | 636ab40d3741e12eaad11d5b50e4b34bfbb258b5 (patch) | |
tree | defaba57dc28c7b5dbe19075e24b6c8c0cefc9b2 /include/armnn/INetwork.hpp | |
parent | 4dc64a69ba383ece509d442598617445a3b4847f (diff) | |
download | armnn-636ab40d3741e12eaad11d5b50e4b34bfbb258b5.tar.gz |
IVGCVSW-3875 Add frontend for SLICE layer
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: Iebe675a0cee02db6f133d48ce58cbc1e233061db
Diffstat (limited to 'include/armnn/INetwork.hpp')
-rw-r--r-- | include/armnn/INetwork.hpp | 10 |
1 files changed, 8 insertions, 2 deletions
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. |