diff options
| author | Giorgio Arena <giorgio.arena@arm.com> | 2018-07-12 15:41:35 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:54:54 +0000 |
| commit | ea55f91e5dd4e5bc766fabbac6df6ce3ab984d0e (patch) | |
| tree | 290272820514c19d9e45b9bc0f0fee4eb3a0bff1 /src/core/CL/cl_kernels/winograd_output_transform.cl | |
| parent | 2df1f99ed6d318c428459fcda8c8052450ccd9bd (diff) | |
| download | ComputeLibrary-ea55f91e5dd4e5bc766fabbac6df6ce3ab984d0e.tar.gz | |
COMPMID-1373 Enable tests for CLWinogradConvolutionLayer for NHWC
Change-Id: I2c6a744f174cfb6c78a9923b737f06537debaa0d
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/139758
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Pablo Tello <pablo.tello@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'src/core/CL/cl_kernels/winograd_output_transform.cl')
| -rw-r--r-- | src/core/CL/cl_kernels/winograd_output_transform.cl | 6 |
1 files changed, 3 insertions, 3 deletions
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; |