diff options
Diffstat (limited to 'src/core/NEON/kernels/NEPermuteKernel.cpp')
| -rw-r--r-- | src/core/NEON/kernels/NEPermuteKernel.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/core/NEON/kernels/NEPermuteKernel.cpp b/src/core/NEON/kernels/NEPermuteKernel.cpp index 897b764b45..2c0db769f5 100644 --- a/src/core/NEON/kernels/NEPermuteKernel.cpp +++ b/src/core/NEON/kernels/NEPermuteKernel.cpp @@ -91,6 +91,7 @@ inline bool is_permutation_supported(const PermutationVector &v) Status validate_arguments(const ITensorInfo *input, const ITensorInfo *output, const PermutationVector &perm) { + ARM_COMPUTE_RETURN_ERROR_ON(input->data_type() == DataType::UNKNOWN); ARM_COMPUTE_RETURN_ERROR_ON_MSG(!is_permutation_supported(perm), "PermutationVector not supported."); const TensorShape output_shape = misc::shape_calculator::compute_permutation_output_shape(*input, perm); |