From aa3fc130acb84f7262fa47d9b5107f0afcf105e0 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Thu, 31 Jan 2019 12:56:02 +0000 Subject: COMPMID-1710: Re-enable winograd on A53. Change-Id: I61e8f77062030067b0680d925b71206cbdac7b62 Signed-off-by: Georgios Pinitas Reviewed-on: https://review.mlplatform.org/611 Tested-by: Arm Jenkins Reviewed-by: Isabella Gottardi Reviewed-by: Pablo Marquez --- src/runtime/NEON/functions/NEConvolutionLayer.cpp | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) (limited to 'src/runtime/NEON/functions') 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; } -- cgit v1.2.1