aboutsummaryrefslogtreecommitdiff
path: root/include/armnn/ILayerVisitor.hpp
diff options
context:
space:
mode:
authorAron 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
commit636ab40d3741e12eaad11d5b50e4b34bfbb258b5 (patch)
treedefaba57dc28c7b5dbe19075e24b6c8c0cefc9b2 /include/armnn/ILayerVisitor.hpp
parent4dc64a69ba383ece509d442598617445a3b4847f (diff)
downloadarmnn-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/ILayerVisitor.hpp')
-rw-r--r--include/armnn/ILayerVisitor.hpp8
1 files changed, 8 insertions, 0 deletions
diff --git a/include/armnn/ILayerVisitor.hpp b/include/armnn/ILayerVisitor.hpp
index a504a4190d..6c0977303c 100644
--- a/include/armnn/ILayerVisitor.hpp
+++ b/include/armnn/ILayerVisitor.hpp
@@ -357,6 +357,14 @@ public:
virtual void VisitRsqrtLayer(const IConnectableLayer* layer,
const char* name = nullptr) = 0;
+ /// Function that a slice layer should call back to when its Accept(ILayerVisitor&) function is invoked.
+ /// @param layer - pointer to the layer which is calling back to this visit function.
+ /// @param sliceDescriptor - SliceDescriptor to configure the slice operation.
+ /// @param name - Optional name for the layer.
+ virtual void VisitSliceLayer(const IConnectableLayer* layer,
+ const SliceDescriptor& sliceDescriptor,
+ const char* name = nullptr) = 0;
+
/// Function that a softmax layer should call back to when its Accept(ILayerVisitor&) function is invoked.
/// @param layer - pointer to the layer which is calling back to this visit function.