aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/layers/ConcatLayer.hpp
diff options
context:
space:
mode:
authorSadik Armagan <sadik.armagan@arm.com>2020-08-04 14:01:05 +0100
committerKeithARM <keith.davis@arm.com>2020-08-07 12:44:19 +0000
commit76615a5edd55b890acdd5fb078d9242e1e719a45 (patch)
tree5c75480b2f7594fa8934952d0037da543026d58c /src/armnn/layers/ConcatLayer.hpp
parente88167264991b8debe56f095abbc262f7266c5d4 (diff)
downloadarmnn-76615a5edd55b890acdd5fb078d9242e1e719a45.tar.gz
IVGCVSW-5108 Allow Concat to use subtensor on x and y
* Updated ConcatLayer to allow using subtensors on x/y if padding is not required Signed-off-by: Sadik Armagan <sadik.armagan@arm.com> Change-Id: I46a8fb9f17b976b76e069bb82614b6628a206717
Diffstat (limited to 'src/armnn/layers/ConcatLayer.hpp')
-rw-r--r--src/armnn/layers/ConcatLayer.hpp2
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnn/layers/ConcatLayer.hpp b/src/armnn/layers/ConcatLayer.hpp
index 84eba2e7c9..eaa5c15a9c 100644
--- a/src/armnn/layers/ConcatLayer.hpp
+++ b/src/armnn/layers/ConcatLayer.hpp
@@ -56,7 +56,7 @@ protected:
private:
template <typename FactoryType>
- void CreateTensors(const FactoryType& factory);
+ void CreateTensors(const TensorHandleFactoryRegistry& registry, const FactoryType& factory);
};