diff options
author | Jim Flynn <jim.flynn@arm.com> | 2019-05-10 13:55:21 +0100 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-05-14 10:35:28 +0000 |
commit | 906f94631aa7ef590b9d8ff45507e818a0d1ac2c (patch) | |
tree | bbb9d805b9424f92fe4f6a7030878e101873b03f /include/armnn/ILayerVisitor.hpp | |
parent | 305620201159eb47e005a46d60f80c6a2c763d51 (diff) | |
download | armnn-906f94631aa7ef590b9d8ff45507e818a0d1ac2c.tar.gz |
IVGCVSW-3076 Add ConcatLayer methods to public API
!android-nn-driver:1120
Change-Id: I5192fa3deb4ea9766d38ad0bf4dfbfa0b4924c41
Signed-off-by: Jim Flynn <jim.flynn@arm.com>
Diffstat (limited to 'include/armnn/ILayerVisitor.hpp')
-rw-r--r-- | include/armnn/ILayerVisitor.hpp | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/include/armnn/ILayerVisitor.hpp b/include/armnn/ILayerVisitor.hpp index ab793bc587..10d0cc6b63 100644 --- a/include/armnn/ILayerVisitor.hpp +++ b/include/armnn/ILayerVisitor.hpp @@ -58,6 +58,20 @@ public: const BatchToSpaceNdDescriptor& batchToSpaceNdDescriptor, const char* name = nullptr) = 0; + /// Function that a concat 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 mergerDescriptor - WindowsDescriptor to configure the concatenation process. Number of Views must be + /// equal to the number of inputs, and their order must match - e.g. first view + /// corresponds to the first input, second view to the second input, etc.... + /// @param name - Optional name for the layer. + virtual void VisitConcatLayer(const IConnectableLayer* layer, + const OriginsDescriptor& mergerDescriptor, + const char* name = nullptr) + { + // default implementation to ease transition while MergerLayer is being deprecated + VisitMergerLayer(layer, mergerDescriptor, name); + } + /// Function a layer with no inputs and a single output, which always corresponds to /// the passed in constant tensor 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. @@ -211,6 +225,7 @@ public: /// the number of inputs, and their order must match - e.g. first view corresponds to /// the first input, second view to the second input, etc.... /// @param name - Optional name for the layer. + // NOTE: this method will be deprecated and replaced by VisitConcatLayer virtual void VisitMergerLayer(const IConnectableLayer* layer, const OriginsDescriptor& mergerDescriptor, const char* name = nullptr) = 0; |