aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/layers/SplitterLayer.hpp
diff options
context:
space:
mode:
authorKeith Davis <keith.davis@arm.com>2020-08-16 23:44:15 +0100
committerTeresaARM <teresa.charlinreyes@arm.com>2020-08-17 14:00:35 +0000
commit3674f1459664f69eaaaae45fe463237835a93d71 (patch)
tree7bf32bfa98cb9b3ac99eb958b980ef8db23247da /src/armnn/layers/SplitterLayer.hpp
parentc53ceec5d901c2fcf3dade4274d7b3f26abdd212 (diff)
downloadarmnn-3674f1459664f69eaaaae45fe463237835a93d71.tar.gz
IVGCVSW-5107 Allow Split to use subtensor on x and y
Signed-off-by: Keith Davis <keith.davis@arm.com> Change-Id: I2370d260b750f36842c23f08e8a00ccf976d0aed
Diffstat (limited to 'src/armnn/layers/SplitterLayer.hpp')
-rw-r--r--src/armnn/layers/SplitterLayer.hpp2
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnn/layers/SplitterLayer.hpp b/src/armnn/layers/SplitterLayer.hpp
index bd20890162..ae725b9ad0 100644
--- a/src/armnn/layers/SplitterLayer.hpp
+++ b/src/armnn/layers/SplitterLayer.hpp
@@ -57,7 +57,7 @@ protected:
private:
template <typename FactoryType>
- void CreateTensors(const FactoryType& factory);
+ void CreateTensors(const TensorHandleFactoryRegistry& registry, const FactoryType& factory);
};
} // namespace