diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/runtime/CL/functions/CLConvolutionLayer.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/runtime/CL/functions/CLConvolutionLayer.cpp b/src/runtime/CL/functions/CLConvolutionLayer.cpp index 1082a475b2..96d7cc72c2 100644 --- a/src/runtime/CL/functions/CLConvolutionLayer.cpp +++ b/src/runtime/CL/functions/CLConvolutionLayer.cpp @@ -232,7 +232,7 @@ ConvolutionMethod CLConvolutionLayer::get_convolution_method(const ITensorInfo * return ConvolutionMethod::FFT; } } - else + else if(is_data_type_float(input->data_type())) { if((weights->dimension(idx_h) >= 5) && (input->dimension(idx_c) >= output->dimension(idx_c)) && (CLDirectConvolutionLayer::validate(input, weights, nullptr, output, conv_info, act_info))) { |