diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-06-27 17:00:52 +0100 |
|---|---|---|
| committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-07-26 11:55:15 +0000 |
| commit | cfa2bba98169cb5ab1945462514be1b6badf7d98 (patch) | |
| tree | 1635e6e9463e9798c7195f0aa71b5df3f2650df1 /src/core/NEON/kernels/arm_gemm/transform.hpp | |
| parent | f59b16f42ef68bde877b70816ffb953d64c8baa3 (diff) | |
| download | ComputeLibrary-cfa2bba98169cb5ab1945462514be1b6badf7d98.tar.gz | |
COMPMID-2178: Update GEMM assembly code.
Perform offset reduction and requantization within the assembly wrapper.
Change-Id: I5d5b3e1f6f9ef4c71805362c57f88ff199c027a3
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1541
Comments-Addressed: Pablo Marquez <pablo.tello@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/NEON/kernels/arm_gemm/transform.hpp')
| -rw-r--r-- | src/core/NEON/kernels/arm_gemm/transform.hpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/core/NEON/kernels/arm_gemm/transform.hpp b/src/core/NEON/kernels/arm_gemm/transform.hpp index 0330783a0b..d790d566b6 100644 --- a/src/core/NEON/kernels/arm_gemm/transform.hpp +++ b/src/core/NEON/kernels/arm_gemm/transform.hpp @@ -23,6 +23,10 @@ */ #pragma once +#include "utils.hpp" + +namespace arm_gemm { + /* * Generic transform. * @@ -80,6 +84,7 @@ struct TransformImpl { *out++ = static_cast<TOut>(0); } } + // "row" tail - row is out of range so fill with zeros always. TOut zeroval = static_cast<TOut>(0); int pads = blank_rows * (fill_cols + blank_cols); @@ -114,3 +119,5 @@ void Transform( /*****************************************************************************/ #include "transforms/list.hpp" + +} // namespace arm_gemm |