diff options
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 |