diff options
| -rw-r--r-- | arm_compute/graph/backends/ValidateHelpers.h | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/arm_compute/graph/backends/ValidateHelpers.h b/arm_compute/graph/backends/ValidateHelpers.h index 237d4ae2a4..c203e8c885 100644 --- a/arm_compute/graph/backends/ValidateHelpers.h +++ b/arm_compute/graph/backends/ValidateHelpers.h @@ -97,12 +97,15 @@ Status validate_convolution_layer(ConvolutionLayerNode &node) case ConvolutionMethod::WINOGRAD: status = WinogradConvolutionLayer::validate(input, weights, biases, output, conv_info); break; + case ConvolutionMethod::DEFAULT: + status = ConvolutionLayer::validate(input, weights, biases, output, conv_info); + break; default: break; } // If validation fails try the Default approach - if(!bool(status) || (conv_algorithm == ConvolutionMethod::DEFAULT)) + if(!bool(status)) { status = ConvolutionLayer::validate(input, weights, biases, output, conv_info); if(bool(status)) |