diff options
| author | Xinghang Zhou <xinghang.zhou@arm.com> | 2017-11-02 16:37:24 +0800 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:42:17 +0000 |
| commit | 4af62a0202bbc103a95a1c2d04c2425418935a0f (patch) | |
| tree | 0800efdcec65a4eee899ca033d432a1082a08147 /src/core/GLES_COMPUTE/cs_shaders/pooling_layer.cs | |
| parent | 7c435f2e32e3441ac6c288e786f25c86b65e1453 (diff) | |
| download | ComputeLibrary-4af62a0202bbc103a95a1c2d04c2425418935a0f.tar.gz | |
APPBROWSER-289 DirectConvolution1*1 optimization for FP16, and fix a typo in PoolingLayer
Change-Id: I9cd3d872e353a9a404ab1d188d0d48a0965c5916
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/112047
Tested-by: BSG Visual Compute Jenkins server to access repositories on http://mpd-gerrit.cambridge.arm.com <bsgcomp@arm.com>
Reviewed-by: Joel Liang <joel.liang@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'src/core/GLES_COMPUTE/cs_shaders/pooling_layer.cs')
| -rw-r--r-- | src/core/GLES_COMPUTE/cs_shaders/pooling_layer.cs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/core/GLES_COMPUTE/cs_shaders/pooling_layer.cs b/src/core/GLES_COMPUTE/cs_shaders/pooling_layer.cs index 1e0fee4688..401b002111 100644 --- a/src/core/GLES_COMPUTE/cs_shaders/pooling_layer.cs +++ b/src/core/GLES_COMPUTE/cs_shaders/pooling_layer.cs @@ -608,8 +608,8 @@ void main(void) // Divide by pool region in case of average pooling int start_x = int(gl_GlobalInvocationID.x) * STRIDE_X - PAD_X; int start_y = int(gl_GlobalInvocationID.y) * STRIDE_Y - PAD_Y; - int end_x = int(min(STRIDE_X + POOL_SIZE, MAX_WIDTH)); - int end_y = int(min(STRIDE_Y + POOL_SIZE, MAX_HEIGHT)); + int end_x = int(min(start_x + POOL_SIZE, MAX_WIDTH)); + int end_y = int(min(start_y + POOL_SIZE, MAX_HEIGHT)); float res1 = float((end_y - start_y) * (end_x - start_x)); res = DIV_OP(res, res1); } |