From c8a85bac94a2d647fa8fdcd6efe9fd7af9ff4d29 Mon Sep 17 00:00:00 2001 From: Sanghoon Lee Date: Wed, 29 Nov 2017 11:23:14 +0000 Subject: COMPMID-563 - Implement reference and CL/NEON validation for Custom_Convolution (Output U8) Change-Id: I57a6db857474929322206ee7440e088bc0bbbbe2 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/111080 Tested-by: BSG Visual Compute Jenkins server to access repositories on http://mpd-gerrit.cambridge.arm.com Reviewed-by: Gian Marco Iodice Reviewed-by: Pablo Tello --- src/core/NEON/kernels/NEConvolutionKernel.cpp | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) (limited to 'src/core/NEON/kernels/NEConvolutionKernel.cpp') diff --git a/src/core/NEON/kernels/NEConvolutionKernel.cpp b/src/core/NEON/kernels/NEConvolutionKernel.cpp index 263fbe058a..3a4884eb21 100644 --- a/src/core/NEON/kernels/NEConvolutionKernel.cpp +++ b/src/core/NEON/kernels/NEConvolutionKernel.cpp @@ -627,16 +627,17 @@ void NEConvolutionKernel::run(const Window &window, const ThreadInf ARM_COMPUTE_ERROR_ON_UNCONFIGURED_KERNEL(this); ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(INEKernel::window(), window); - switch(_output->info()->format()) + switch(_output->info()->data_type()) { - case Format::U8: + case DataType::U8: convolution(window); break; - case Format::S16: + case DataType::S16: convolution(window); break; default: - ARM_COMPUTE_ERROR("Not supported"); + ARM_COMPUTE_ERROR("Not supported Data type!"); + break; } } -- cgit v1.2.1