diff options
Diffstat (limited to 'src/cpu/kernels/select/generic/neon/integer.cpp')
-rw-r--r-- | src/cpu/kernels/select/generic/neon/integer.cpp | 40 |
1 files changed, 26 insertions, 14 deletions
diff --git a/src/cpu/kernels/select/generic/neon/integer.cpp b/src/cpu/kernels/select/generic/neon/integer.cpp index 71b2f0b933..135087c261 100644 --- a/src/cpu/kernels/select/generic/neon/integer.cpp +++ b/src/cpu/kernels/select/generic/neon/integer.cpp @@ -25,59 +25,71 @@ #include "arm_compute/core/TensorInfo.h" #include "arm_compute/core/Types.h" -#include <arm_neon.h> - #include "src/cpu/kernels/select/generic/neon/impl.h" +#include <arm_neon.h> + namespace arm_compute { namespace cpu { -void neon_s8_select_same_rank(const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) +void neon_s8_select_same_rank( + const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) { return select_op_8<int8_t, uint8x16_t>(c, x, y, output, window); } -void neon_s16_select_same_rank(const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) +void neon_s16_select_same_rank( + const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) { return select_op_16<int16_t, uint16x8_t>(c, x, y, output, window); } -void neon_s32_select_same_rank(const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) +void neon_s32_select_same_rank( + const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) { return select_op_32<int32_t, uint32x4_t>(c, x, y, output, window); } -void neon_s8_select_not_same_rank(const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) +void neon_s8_select_not_same_rank( + const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) { return select_op_not_same_rank<int8_t>(c, x, y, output, window); } -void neon_s16_select_not_same_rank(const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) +void neon_s16_select_not_same_rank( + const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) { return select_op_not_same_rank<int16_t>(c, x, y, output, window); } -void neon_s32_select_not_same_rank(const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) +void neon_s32_select_not_same_rank( + const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) { return select_op_not_same_rank<int32_t>(c, x, y, output, window); } -void neon_u8_select_same_rank(const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) +void neon_u8_select_same_rank( + const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) { return select_op_8<uint8_t, uint8x16_t>(c, x, y, output, window); } -void neon_u16_select_same_rank(const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) +void neon_u16_select_same_rank( + const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) { return select_op_16<uint16_t, uint16x8_t>(c, x, y, output, window); } -void neon_u32_select_same_rank(const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) +void neon_u32_select_same_rank( + const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) { return select_op_32<uint32_t, uint32x4_t>(c, x, y, output, window); } -void neon_u8_select_not_same_rank(const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) +void neon_u8_select_not_same_rank( + const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) { return select_op_not_same_rank<uint8_t>(c, x, y, output, window); } -void neon_u16_select_not_same_rank(const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) +void neon_u16_select_not_same_rank( + const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) { return select_op_not_same_rank<uint16_t>(c, x, y, output, window); } -void neon_u32_select_not_same_rank(const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) +void neon_u32_select_not_same_rank( + const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output, const Window &window) { return select_op_not_same_rank<uint32_t>(c, x, y, output, window); } |