diff options
| -rw-r--r-- | src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp b/src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp index c69af55f57..190487eced 100644 --- a/src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp +++ b/src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp @@ -839,7 +839,7 @@ Status validate_arguments(const ITensorInfo *mm_result, const ITensorInfo *vecto } // Check Tensor Rank of vector_sum_row - ARM_COMPUTE_RETURN_ERROR_ON(vector_sum_row->num_dimensions() > 2); + ARM_COMPUTE_RETURN_ERROR_ON(vector_sum_row->num_dimensions() > 3); } if(output->total_size() != 0) |