diff options
Diffstat (limited to 'src/runtime/NEON/functions/NEWidthConcatenateLayer.cpp')
-rw-r--r-- | src/runtime/NEON/functions/NEWidthConcatenateLayer.cpp | 27 |
1 files changed, 24 insertions, 3 deletions
diff --git a/src/runtime/NEON/functions/NEWidthConcatenateLayer.cpp b/src/runtime/NEON/functions/NEWidthConcatenateLayer.cpp index 7e435c34b1..17c352b8f3 100644 --- a/src/runtime/NEON/functions/NEWidthConcatenateLayer.cpp +++ b/src/runtime/NEON/functions/NEWidthConcatenateLayer.cpp @@ -40,7 +40,8 @@ NEWidthConcatenateLayer::NEWidthConcatenateLayer() { } -Status NEWidthConcatenateLayer::validate(const std::vector<ITensorInfo *> &inputs_vector, const ITensorInfo *output) +template <typename TensorInfoType, typename> +inline Status NEWidthConcatenateLayer::validate_internal(const std::vector<TensorInfoType *> &inputs_vector, const ITensorInfo *output) { ARM_COMPUTE_RETURN_ERROR_ON_NULLPTR(output); ARM_COMPUTE_RETURN_ERROR_ON(inputs_vector.size() < 2); @@ -60,8 +61,8 @@ Status NEWidthConcatenateLayer::validate(const std::vector<ITensorInfo *> &input return Status{}; } - -void NEWidthConcatenateLayer::configure(std::vector<ITensor *> inputs_vector, ITensor *output) +template <typename TensorType, typename> +inline void NEWidthConcatenateLayer::configure_internal(std::vector<TensorType *> &&inputs_vector, ITensor *output) { _num_inputs = inputs_vector.size(); @@ -87,6 +88,26 @@ void NEWidthConcatenateLayer::configure(std::vector<ITensor *> inputs_vector, IT } } +void NEWidthConcatenateLayer::configure(std::vector<ITensor *> inputs_vector, ITensor *output) +{ + configure_internal(std::move(inputs_vector), output); +} + +void NEWidthConcatenateLayer::configure(std::vector<const ITensor *> inputs_vector, ITensor *output) +{ + configure_internal(std::move(inputs_vector), output); +} + +Status NEWidthConcatenateLayer::validate(const std::vector<ITensorInfo *> &inputs_vector, const ITensorInfo *output) +{ + return validate_internal(inputs_vector, output); +} + +Status NEWidthConcatenateLayer::validate(const std::vector<const ITensorInfo *> &inputs_vector, const ITensorInfo *output) +{ + return validate_internal(inputs_vector, output); +} + void NEWidthConcatenateLayer::run() { for(unsigned i = 0; i < _num_inputs; ++i) |