From 9f15c5e838085294ce391d3bdcf5ec7752650ea8 Mon Sep 17 00:00:00 2001
From: Michalis Spyrou <michalis.spyrou@arm.com>
Date: Wed, 3 Apr 2019 19:48:54 +0100
Subject: COMPMID-2062 Rework NEON ConcatenateLayer

Change-Id: I2703c99f651e4f06c4e44e39a85a8a8d201c5362
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/944
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
---
 arm_compute/runtime/NEON/functions/NEConcatenateLayer.h | 10 +++-------
 1 file changed, 3 insertions(+), 7 deletions(-)

(limited to 'arm_compute/runtime/NEON/functions/NEConcatenateLayer.h')

diff --git a/arm_compute/runtime/NEON/functions/NEConcatenateLayer.h b/arm_compute/runtime/NEON/functions/NEConcatenateLayer.h
index b70d6ebc7c..7dfbcf9199 100644
--- a/arm_compute/runtime/NEON/functions/NEConcatenateLayer.h
+++ b/arm_compute/runtime/NEON/functions/NEConcatenateLayer.h
@@ -77,13 +77,9 @@ public:
     void run() override;
 
 private:
-    void configure_h_concatenate(std::vector<ITensor *> inputs_vector, ITensor *output);
-    static Status validate_h_concatenate(const std::vector<ITensorInfo *> &inputs_vector, const ITensorInfo *output);
-
-    std::unique_ptr<IFunction>                        _concat_function;
-    std::unique_ptr<NEHeightConcatenateLayerKernel[]> _hconcat_kernels;
-    unsigned int                                      _num_inputs;
-    unsigned int                                      _axis;
+    std::vector<std::unique_ptr<INEKernel>> _concat_kernels;
+    unsigned int                            _num_inputs;
+    unsigned int                            _axis;
 };
 } // namespace arm_compute
 #endif /* __ARM_COMPUTE_NECONCATENATELAYER_H__ */
-- 
cgit v1.2.1