diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2020-03-04 15:44:46 +0000 |
---|---|---|
committer | Michalis Spyrou <michalis.spyrou@arm.com> | 2020-03-04 17:09:56 +0000 |
commit | 773e43cbe7c8934ec7b6b9413a8b9ea13b44c756 (patch) | |
tree | 87ffc557e466bbaa66fadf2a5838844ea6c913c8 | |
parent | c68c217b8fe1c157ba141512ff82bcae08401f47 (diff) | |
download | ComputeLibrary-773e43cbe7c8934ec7b6b9413a8b9ea13b44c756.tar.gz |
COMPMID-3069: Add NERNN missing assert
Change-Id: I1ce028fa3ea3032a2a66b6d7de24aca3f4e5598a
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/2830
Reviewed-by: Pablo Marquez <pablo.tello@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
-rw-r--r-- | src/runtime/NEON/functions/NERNNLayer.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/runtime/NEON/functions/NERNNLayer.cpp b/src/runtime/NEON/functions/NERNNLayer.cpp index 67f4064632..154b060c3d 100644 --- a/src/runtime/NEON/functions/NERNNLayer.cpp +++ b/src/runtime/NEON/functions/NERNNLayer.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018-2019 ARM Limited. + * Copyright (c) 2018-2020 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -47,6 +47,7 @@ Status NERNNLayer::validate(const ITensorInfo *input, const ITensorInfo *weights const int idx_width = get_data_layout_dimension_index(input->data_layout(), DataLayoutDimension::WIDTH); const int idx_height = get_data_layout_dimension_index(input->data_layout(), DataLayoutDimension::HEIGHT); ARM_COMPUTE_RETURN_ERROR_ON(input->dimension(idx_width) != weights->dimension(idx_width)); + ARM_COMPUTE_RETURN_ERROR_ON(input->num_dimensions() != 2); ARM_COMPUTE_RETURN_ERROR_ON(weights->dimension(idx_height) != recurrent_weights->dimension(idx_width)); ARM_COMPUTE_RETURN_ERROR_ON(recurrent_weights->dimension(idx_width) != recurrent_weights->dimension(idx_height)); ARM_COMPUTE_RETURN_ERROR_ON(bias->num_dimensions() != 1); |