diff options
| author | Alexander Jung <a.jung@uni-tuebingen.de> | 2020-08-05 16:26:41 +0200 |
|---|---|---|
| committer | Alexander Jung <a.jung@uni-tuebingen.de> | 2020-08-06 13:57:40 +0000 |
| commit | 187558fdacc457887d21b61f50425cd55187722a (patch) | |
| tree | 7de2aaa86a72b40c68e25a76132dbfffc21c0f68 | |
| parent | dd4cfc0c82a7e559e9a1985e5616f5f3c9d1c299 (diff) | |
| download | ComputeLibrary-187558fdacc457887d21b61f50425cd55187722a.tar.gz | |
Added missing parameter num_groups to the validate call of NEConvolutionLayer.
Change-Id: I5f14b9175b0d72133536578c1d019cab98b2f746
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3679
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Pablo Marquez <pablo.tello@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
| -rw-r--r-- | src/runtime/NEON/functions/NEConvolutionLayer.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/runtime/NEON/functions/NEConvolutionLayer.cpp b/src/runtime/NEON/functions/NEConvolutionLayer.cpp index c6798913c7..491425c487 100644 --- a/src/runtime/NEON/functions/NEConvolutionLayer.cpp +++ b/src/runtime/NEON/functions/NEConvolutionLayer.cpp @@ -48,7 +48,7 @@ void NEConvolutionLayer::configure(ITensor *input, const ITensor *weights, const ARM_COMPUTE_ERROR_ON_NULLPTR(input, weights, output); ARM_COMPUTE_UNUSED(num_groups); ARM_COMPUTE_ERROR_THROW_ON(NEConvolutionLayer::validate(input->info(), weights->info(), ((biases != nullptr) ? biases->info() : nullptr), output->info(), conv_info, weights_info, dilation, act_info, - enable_fast_math)); + enable_fast_math, num_groups)); switch(NEConvolutionLayer::get_convolution_method(input->info(), weights->info(), output->info(), conv_info, weights_info, dilation, act_info, enable_fast_math)) { |