diff options
author | zhenglin <zhenglin.li@arm.com> | 2017-12-06 16:41:20 +0800 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:42:17 +0000 |
commit | 4f7f255c0e8bc1e3a910e0942f42bf5a8fd21311 (patch) | |
tree | 978a8a181108196078e6cfd2f2fadb5c87bfa5cb /src/core/GLES_COMPUTE/cs_shaders/helpers_cs.h | |
parent | b9d38ee6378f3035f8dbad442223d3d9e2f3dc4f (diff) | |
download | ComputeLibrary-4f7f255c0e8bc1e3a910e0942f42bf5a8fd21311.tar.gz |
APPBROWSER-332 Code refactoring for batchnormalization_layer.cs
Change-Id: Ib695e7551994a10355c823840d3fb6237aef0a65
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/112054
Reviewed-by: Joel Liang <joel.liang@arm.com>
Tested-by: BSG Visual Compute Jenkins server to access repositories on http://mpd-gerrit.cambridge.arm.com <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'src/core/GLES_COMPUTE/cs_shaders/helpers_cs.h')
-rwxr-xr-x[-rw-r--r--] | src/core/GLES_COMPUTE/cs_shaders/helpers_cs.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/core/GLES_COMPUTE/cs_shaders/helpers_cs.h b/src/core/GLES_COMPUTE/cs_shaders/helpers_cs.h index 31deaf67e3..404b46a89d 100644..100755 --- a/src/core/GLES_COMPUTE/cs_shaders/helpers_cs.h +++ b/src/core/GLES_COMPUTE/cs_shaders/helpers_cs.h @@ -222,6 +222,9 @@ ImageIterator update_image_from_tensor3D_iter_offset(uint element_shift, uint of #define TENSOR3D_OFFSET(tensor_iter, x, y, z) \ uint(tensor3D_offset_in_bytes(tensor_iter, int(x), int(y), int(z)) >> tensor_iter.element_shift) +#define TENSOR_OFFSET_ADVANCE_IN_BYTES(tensor_iter, n) \ + uint((tensor_iter.current_offset_in_bytes + int(n)) >> tensor_iter.element_shift) + #define CURRENT_ITEM_OFFSET(tensor_iter) \ uint(tensor_iter.current_offset_in_bytes >> tensor_iter.element_shift) |