commit 8c571692a8236be8605a753e231d240094428be5
author Michalis Spyrou <michalis.spyrou@arm.com> Fri Apr 05 11:29:52 2019 +0100
committer Michalis Spyrou <michalis.spyrou@arm.com> Mon Apr 08 13:49:12 2019 +0000
tree abb38e31254373fcf4c2c77d5c03073073d92c13
parent da953f2e17b401ff595ac88008cd5e90440ebf74

COMPMID-2062 Rework CL ConcatenateLayer

Change-Id: I5a60cd0e822b8912132a6785057921bbf6ef8f7f
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/951
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/CL/functions/CLConcatenateLayer.h

diff --git a/arm_compute/runtime/CL/functions/CLConcatenateLayer.h b/arm_compute/runtime/CL/functions/CLConcatenateLayer.h
index 25a0153..5cf09c8 100644
--- a/arm_compute/runtime/CL/functions/CLConcatenateLayer.h
+++ b/arm_compute/runtime/CL/functions/CLConcatenateLayer.h
@@ -77,13 +77,9 @@
     void run() override;
 
 private:
-    void configure_h_concatenate(std::vector<ICLTensor *> inputs_vector, ICLTensor *output);
-    static Status validate_h_concatenate(const std::vector<ITensorInfo *> &inputs_vector, const ITensorInfo *output);
-
-    std::unique_ptr<IFunction>                        _concat_function;
-    std::unique_ptr<CLHeightConcatenateLayerKernel[]> _hconcat_kernels;
-    unsigned int                                      _num_inputs;
-    unsigned int                                      _axis;
+    std::vector<std::unique_ptr<ICLKernel>> _concat_kernels;
+    unsigned int                            _num_inputs;
+    unsigned int                            _axis;
 };
 }
 #endif /* __ARM_COMPUTE_CLCONCATENATELAYER_H__ */