aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/CL/kernels/CLPermuteKernel.h
diff options
context:
space:
mode:
authorMichele Di Giorgio <michele.digiorgio@arm.com>2018-02-02 15:58:01 +0000
committerAnthony Barbier <anthony.barbier@arm.com>2018-11-02 16:45:42 +0000
commitb28f329d0cdc2e5f2f72fd3ecc1913c6b913aee8 (patch)
tree7264618fb656649b3e1b206f0b71ac634a94c066 /arm_compute/core/CL/kernels/CLPermuteKernel.h
parent2d40555fff73ac8c23b9a21c3b4fc2a9fa01eb2e (diff)
downloadComputeLibrary-b28f329d0cdc2e5f2f72fd3ecc1913c6b913aee8.tar.gz
COMPMID-754: Add validation method to CLPermute kernel
Change-Id: If6f3888a035b557a6c369efa22b56d6c8d3efbd3 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/118789 Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com> Tested-by: Jenkins <bsgcomp@arm.com> Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
Diffstat (limited to 'arm_compute/core/CL/kernels/CLPermuteKernel.h')
-rw-r--r--arm_compute/core/CL/kernels/CLPermuteKernel.h9
1 files changed, 9 insertions, 0 deletions
diff --git a/arm_compute/core/CL/kernels/CLPermuteKernel.h b/arm_compute/core/CL/kernels/CLPermuteKernel.h
index 8f96529d32..b01df64ebd 100644
--- a/arm_compute/core/CL/kernels/CLPermuteKernel.h
+++ b/arm_compute/core/CL/kernels/CLPermuteKernel.h
@@ -54,6 +54,15 @@ public:
* @param[in] perm Permutation vector
*/
void configure(const ICLTensor *input, ICLTensor *output, const PermutationVector &perm);
+ /** Static function to check if given info will lead to a valid configuration of @ref CLPermuteKernel
+ *
+ * @param[in] input First tensor input info. Data types supported: U8/S8/QS8/QASYMM8/U16/S16/QS16/F16/U32/S32/F32.
+ * @param[in] output Output tensor info. Data types supported: same as @p input.
+ * @param[in] perm Permutation vector
+ *
+ * @return a status
+ */
+ static Status validate(const ITensorInfo *input, const ITensorInfo *output, const PermutationVector &perm);
// Inherited methods overridden:
void run(const Window &window, cl::CommandQueue &queue) override;