aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/runtime/NEON/functions/NEConcatenateLayer.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEConcatenateLayer.h')
-rw-r--r--arm_compute/runtime/NEON/functions/NEConcatenateLayer.h10
1 files changed, 3 insertions, 7 deletions
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__ */