From ea55f91e5dd4e5bc766fabbac6df6ce3ab984d0e Mon Sep 17 00:00:00 2001 From: Giorgio Arena Date: Thu, 12 Jul 2018 15:41:35 +0100 Subject: COMPMID-1373 Enable tests for CLWinogradConvolutionLayer for NHWC Change-Id: I2c6a744f174cfb6c78a9923b737f06537debaa0d Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/139758 Tested-by: Jenkins Reviewed-by: Pablo Tello Reviewed-by: Anthony Barbier --- src/core/CL/cl_kernels/winograd_output_transform.cl | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'src/core/CL/cl_kernels/winograd_output_transform.cl') diff --git a/src/core/CL/cl_kernels/winograd_output_transform.cl b/src/core/CL/cl_kernels/winograd_output_transform.cl index bb159a7f4c..7065da1cc5 100644 --- a/src/core/CL/cl_kernels/winograd_output_transform.cl +++ b/src/core/CL/cl_kernels/winograd_output_transform.cl @@ -507,7 +507,7 @@ __kernel void winograd_output_transform_4x4_3x3_nhwc( // Add bias Vector bias = CONVERT_TO_VECTOR_STRUCT_NO_STEP(bias); - float b = (float) * ((__global float *)(vector_offset(&bias, z_out))); + float b = (float) * ((__global float *)(vector_offset(&bias, x_out))); out00 += (float)b; out01 += (float)b; @@ -856,7 +856,7 @@ __kernel void winograd_output_transform_4x4_5x5_nhwc( // Add bias Vector bias = CONVERT_TO_VECTOR_STRUCT_NO_STEP(bias); - float b = (float) * ((__global float *)(vector_offset(&bias, z_out))); + float b = (float) * ((__global float *)(vector_offset(&bias, x_out))); out00 += (float)b; out01 += (float)b; @@ -981,7 +981,7 @@ __kernel void winograd_output_transform_4x4_5x5_nhwc( // Add bias Vector bias = CONVERT_TO_VECTOR_STRUCT_NO_STEP(bias); - float b = (float) * ((__global float *)(vector_offset(&bias, z_out))); + float b = (float) * ((__global float *)(vector_offset(&bias, x_out))); out_col0 += (float4)b; out_col1 += (float4)b; -- cgit v1.2.1