38 template <
typename ScalarType>
42 const auto all_true_pg = wrapper::svptrue<ScalarType>();
43 const auto window_start_x =
static_cast<int>(window.
x().
start());
44 const auto window_end_x =
static_cast<int>(window.
x().
end());
60 if (is_broadcast_across_x)
62 const bool is_broadcast_input_2 = input2_win.
x().
step() == 0;
63 Window broadcast_win = is_broadcast_input_2 ? input2_win : input1_win;
64 Window non_broadcast_win = !is_broadcast_input_2 ? input2_win : input1_win;
65 const ITensor *broadcast_tensor = is_broadcast_input_2 ? src1 : src0;
66 const ITensor *non_broadcast_tensor = !is_broadcast_input_2 ? src1 : src0;
71 Iterator broadcast_input(broadcast_tensor, broadcast_win);
72 Iterator non_broadcast_input(non_broadcast_tensor, non_broadcast_win);
79 const auto non_broadcast_input_ptr =
reinterpret_cast<const ScalarType *
>(non_broadcast_input.
ptr());
80 const auto output_ptr =
reinterpret_cast<ScalarType *
>(output.
ptr());
82 const ScalarType broadcast_value = *
reinterpret_cast<const ScalarType *
>(broadcast_input.
ptr());
83 const auto broadcast_value_vec = wrapper::svdup_n(broadcast_value);
85 int x = window_start_x;
86 svbool_t pg = wrapper::svwhilelt<ScalarType>(x, window_end_x);
89 const auto non_broadcast_v = svld1(pg, non_broadcast_input_ptr + x);
90 auto res = is_sat ? wrapper::svqadd(broadcast_value_vec, non_broadcast_v)
91 : svadd_z(pg, broadcast_value_vec, non_broadcast_v);
92 svst1(pg, output_ptr + x, res);
94 x += wrapper::svcnt<ScalarType>();
95 pg = wrapper::svwhilelt<ScalarType>(x, window_end_x);
96 }
while (svptest_any(all_true_pg, pg));
98 broadcast_input, non_broadcast_input, output);
114 const auto input1_ptr =
reinterpret_cast<const ScalarType *
>(input1.
ptr());
115 const auto input2_ptr =
reinterpret_cast<const ScalarType *
>(input2.
ptr());
116 const auto output_ptr =
reinterpret_cast<ScalarType *
>(output.
ptr());
118 int x = window_start_x;
119 svbool_t pg = wrapper::svwhilelt<ScalarType>(x, window_end_x);
122 const auto val1 = svld1(pg, input1_ptr + x);
123 const auto val2 = svld1(pg, input2_ptr + x);
124 const auto res = is_sat ? wrapper::svqadd(val1, val2) : svadd_z(pg, val1, val2);
125 svst1(pg, output_ptr + x, res);
127 x += wrapper::svcnt<ScalarType>();
128 pg = wrapper::svwhilelt<ScalarType>(x, window_end_x);
129 }
while (svptest_any(all_true_pg, pg));
131 input1, input2, output);
142 #if defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC) && defined(ENABLE_FP16_KERNELS)
143 template void add_same_sve<float16_t>(