diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-06-11 14:14:03 +0100 |
---|---|---|
committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-06-12 08:53:33 +0000 |
commit | 972af1568372f243f241b58fa07f0d3a7f991d1e (patch) | |
tree | 7f00157cc66bef8ced30a8e0a0f5e5305a82111f /include/armnn/ILayerVisitor.hpp | |
parent | e9e1878cb94e45ba71303a8ceda5bc523997fde4 (diff) | |
download | armnn-972af1568372f243f241b58fa07f0d3a7f991d1e.tar.gz |
IVGCVSW-3258 Add front end support for new SpaceToDepth layer
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: Id677e29a734f2b36483d939ad370079bdc11551e
Diffstat (limited to 'include/armnn/ILayerVisitor.hpp')
-rw-r--r-- | include/armnn/ILayerVisitor.hpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/include/armnn/ILayerVisitor.hpp b/include/armnn/ILayerVisitor.hpp index cbddb2d9ac..ab83dbfb05 100644 --- a/include/armnn/ILayerVisitor.hpp +++ b/include/armnn/ILayerVisitor.hpp @@ -336,6 +336,14 @@ public: const SpaceToBatchNdDescriptor& spaceToBatchNdDescriptor, const char* name = nullptr) = 0; + /// Function a space to depth 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 spaceToDepthDescriptor - Parameters for the space to depth operation. + /// @param name - Optional name for the layer. + virtual void VisitSpaceToDepthLayer(const IConnectableLayer* layer, + const SpaceToDepthDescriptor& spaceToDepthDescriptor, + const char* name = nullptr) = 0; + /// Function that a splitter 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 splitterDescriptor - ViewsDescriptor to configure the splitting process. |