diff options
| author | Kurtis Charnock <kurtis.charnock@arm.com> | 2019-11-29 11:43:05 +0000 |
|---|---|---|
| committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-01-14 10:44:34 +0000 |
| commit | ab709a0ea0f7f5c8e02c315afffc300e09c783a8 (patch) | |
| tree | 0894f9e2cbf6f3abc2b6e2f1bc3238253b8f203b /arm_compute | |
| parent | 9a9440ebc0d6710d832abd1ba5daea5af08a5e5c (diff) | |
| download | ComputeLibrary-ab709a0ea0f7f5c8e02c315afffc300e09c783a8.tar.gz | |
COMPMID-2727: Add support for split sizes in NESplit
Signed-off-by: Kurtis Charnock <kurtis.charnock@arm.com>
Change-Id: I14cb2711f3a02bd5f50976cb78fe5865e2062891
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/c/VisualCompute/ComputeLibrary/+/214133
Tested-by: bsgcomp <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/c/2433
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute')
| -rw-r--r-- | arm_compute/runtime/NEON/functions/NESplit.h | 37 |
1 files changed, 4 insertions, 33 deletions
diff --git a/arm_compute/runtime/NEON/functions/NESplit.h b/arm_compute/runtime/NEON/functions/NESplit.h index e4d62048e6..69aef793d5 100644 --- a/arm_compute/runtime/NEON/functions/NESplit.h +++ b/arm_compute/runtime/NEON/functions/NESplit.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018-2019 ARM Limited. + * Copyright (c) 2018-2020 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -24,8 +24,10 @@ #ifndef ARM_COMPUTE_NESPLIT_H #define ARM_COMPUTE_NESPLIT_H +#include "arm_compute/core/ITensor.h" #include "arm_compute/core/Types.h" +#include "arm_compute/runtime/CPP/functions/CPPSplit.h" #include "arm_compute/runtime/IFunction.h" #include "arm_compute/runtime/NEON/functions/NESlice.h" @@ -34,43 +36,12 @@ namespace arm_compute { -// Forward declarations -class ITensor; - /** Basic function to split a tensor along a given axis */ -class NESplit : public IFunction +class NESplit : public CPPSplit<NESlice> { public: - /** Default constructor */ - NESplit(); - /** Initialise the kernel's input and outputs. - * - * @param[in] input The input tensor. Data types supported: All - * @param[out] outputs A vector containing the output tensors. Data types supported: Same as @p input. - * The output tensors should match the input tensor dimensions for all shape dimensions apart - * from the split dimension. - * @param[in] axis Axis on which to split the input. - */ - void configure(const ITensor *input, const std::vector<ITensor *> &outputs, unsigned int axis); - /** Static function to check if given info will lead to a valid configuration of @ref NESplit - * - * @param[in] input The input tensor info. Data types supported: All - * @param[in] outputs A vector containing the output tensors' info. Data types supported: Same as @p input. - * The output tensors should match the input tensor dimensions for all shape dimensions apart - * from the split dimension - * @param[in] axis Axis on which to split the input. - * - * @return a status - */ - static Status validate(const ITensorInfo *input, const std::vector<ITensorInfo *> &outputs, unsigned int axis); - // Inherited methods overridden: void run() override; - -private: - std::vector<ITensor *> _outputs_vector; - std::vector<NESlice> _slice_functions; - unsigned int _num_outputs; }; } // namespace arm_compute #endif /* ARM_COMPUTE_NESPLIT_H */ |