diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-01-31 12:56:02 +0000 |
|---|---|---|
| committer | Pablo Marquez <pablo.tello@arm.com> | 2019-02-01 10:09:35 +0000 |
| commit | aa3fc130acb84f7262fa47d9b5107f0afcf105e0 (patch) | |
| tree | b183d3639f16989c9bb3b856624562403ea04504 /src/runtime/NEON/functions | |
| parent | 6d109965f3641056bb8164dc8450a7327e76e939 (diff) | |
| download | ComputeLibrary-aa3fc130acb84f7262fa47d9b5107f0afcf105e0.tar.gz | |
COMPMID-1710: Re-enable winograd on A53.
Change-Id: I61e8f77062030067b0680d925b71206cbdac7b62
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/611
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Isabella Gottardi <isabella.gottardi@arm.com>
Reviewed-by: Pablo Marquez <pablo.tello@arm.com>
Diffstat (limited to 'src/runtime/NEON/functions')
| -rw-r--r-- | src/runtime/NEON/functions/NEConvolutionLayer.cpp | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/src/runtime/NEON/functions/NEConvolutionLayer.cpp b/src/runtime/NEON/functions/NEConvolutionLayer.cpp index 931e5db65b..5059162032 100644 --- a/src/runtime/NEON/functions/NEConvolutionLayer.cpp +++ b/src/runtime/NEON/functions/NEConvolutionLayer.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2018 ARM Limited. + * Copyright (c) 2017-2019 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -148,8 +148,7 @@ ConvolutionMethod NEConvolutionLayer::get_convolution_method(const ITensorInfo * return (*found).second; } - if(dilation != Size2D(1U, 1U) || Scheduler::get().cpu_info().get_cpu_model() == CPUModel::A53 - || input->dimension(get_data_layout_dimension_index(input->data_layout(), DataLayoutDimension::CHANNEL)) <= 16) + if(dilation != Size2D(1U, 1U) || input->dimension(get_data_layout_dimension_index(input->data_layout(), DataLayoutDimension::CHANNEL)) <= 16) { return ConvolutionMethod::GEMM; } |