38 template <
typename ScalarType>
46 const auto scale_beta_vec = svdup_n_f32(scale_beta);
51 const auto all_true_pg = wrapper::svptrue<ScalarType>();
54 const int inc_1 =
static_cast<int>(svcntw());
55 const int inc_2 =
static_cast<int>(2 * svcntw());
56 const int inc_3 =
static_cast<int>(3 * svcntw());
63 const auto in_ptr =
reinterpret_cast<const ScalarType *
>(in_it.
ptr()) + start_x;
64 const auto out_ptr =
reinterpret_cast<ScalarType *
>(out_it.
ptr()) + start_x;
65 const auto tmp_ptr =
reinterpret_cast<float *
>(tmp);
72 const auto max_val = *
reinterpret_cast<const ScalarType *
>(max_it.
ptr());
73 const auto vec_max = wrapper::svdup_n(max_val);
76 auto vec_sum_0 = svdup_n_f32(0.f);
77 auto vec_sum_1 = svdup_n_f32(0.f);
78 auto vec_sum_2 = svdup_n_f32(0.f);
79 auto vec_sum_3 = svdup_n_f32(0.f);
83 svbool_t pg = wrapper::svwhilelt<ScalarType>(x, input_width);
84 svbool_t pg_0 = svunpklo(svunpklo(pg));
85 svbool_t pg_1 = svunpkhi(svunpklo(pg));
86 svbool_t pg_2 = svunpklo(svunpkhi(pg));
87 svbool_t pg_3 = svunpkhi(svunpkhi(pg));
90 const auto vec_elements = svld1(pg, in_ptr + x);
91 const auto vec_elements_sub = svreinterpret_u8(svsub_z(pg, vec_max, vec_elements));
93 auto vec_elements_flt_0 = svcvt_f32_z(pg_0, svunpklo(svunpklo(vec_elements_sub)));
94 auto vec_elements_flt_1 = svcvt_f32_z(pg_1, svunpkhi(svunpklo(vec_elements_sub)));
95 auto vec_elements_flt_2 = svcvt_f32_z(pg_2, svunpklo(svunpkhi(vec_elements_sub)));
96 auto vec_elements_flt_3 = svcvt_f32_z(pg_3, svunpkhi(svunpkhi(vec_elements_sub)));
100 vec_elements_flt_0 = svmul_f32_z(pg_0, vec_elements_flt_0, scale_beta_vec);
101 vec_elements_flt_1 = svmul_f32_z(pg_1, vec_elements_flt_1, scale_beta_vec);
102 vec_elements_flt_2 = svmul_f32_z(pg_2, vec_elements_flt_2, scale_beta_vec);
103 vec_elements_flt_3 = svmul_f32_z(pg_3, vec_elements_flt_3, scale_beta_vec);
104 vec_sum_0 = svadd_f32_m(pg_0, vec_sum_0, svexp_f32_z(pg_0, vec_elements_flt_0));
105 vec_sum_1 = svadd_f32_m(pg_1, vec_sum_1, svexp_f32_z(pg_1, vec_elements_flt_1));
106 vec_sum_2 = svadd_f32_m(pg_2, vec_sum_2, svexp_f32_z(pg_2, vec_elements_flt_2));
107 vec_sum_3 = svadd_f32_m(pg_3, vec_sum_3, svexp_f32_z(pg_3, vec_elements_flt_3));
111 vec_elements_flt_0 = svexp_f32_z(pg_0, svmul_f32_z(pg_0, vec_elements_flt_0, scale_beta_vec));
112 vec_elements_flt_1 = svexp_f32_z(pg_1, svmul_f32_z(pg_1, vec_elements_flt_1, scale_beta_vec));
113 vec_elements_flt_2 = svexp_f32_z(pg_2, svmul_f32_z(pg_2, vec_elements_flt_2, scale_beta_vec));
114 vec_elements_flt_3 = svexp_f32_z(pg_3, svmul_f32_z(pg_3, vec_elements_flt_3, scale_beta_vec));
115 vec_sum_0 = svadd_f32_m(pg_0, vec_sum_0, vec_elements_flt_0);
116 vec_sum_1 = svadd_f32_m(pg_1, vec_sum_1, vec_elements_flt_1);
117 vec_sum_2 = svadd_f32_m(pg_2, vec_sum_2, vec_elements_flt_2);
118 vec_sum_3 = svadd_f32_m(pg_3, vec_sum_3, vec_elements_flt_3);
121 svst1_f32(pg_0, tmp_ptr + x, vec_elements_flt_0);
122 svst1_f32(pg_1, tmp_ptr + x + inc_1, vec_elements_flt_1);
123 svst1_f32(pg_2, tmp_ptr + x + inc_2, vec_elements_flt_2);
124 svst1_f32(pg_3, tmp_ptr + x + inc_3, vec_elements_flt_3);
126 x += wrapper::svcnt<ScalarType>();
127 pg = wrapper::svwhilelt<ScalarType>(x, input_width);
128 pg_0 = svunpklo(svunpklo(pg));
129 pg_1 = svunpkhi(svunpklo(pg));
130 pg_2 = svunpklo(svunpkhi(pg));
131 pg_3 = svunpkhi(svunpkhi(pg));
132 }
while (svptest_any(all_true_pg, pg));
135 const auto vec_sum = svadd_f32_z(all_true_pg, svadd_f32_z(all_true_pg, vec_sum_0, vec_sum_1),
136 svadd_f32_z(all_true_pg, vec_sum_2, vec_sum_3));
137 sum = svaddv_f32(all_true_pg, vec_sum);
153 constexpr
bool is_qasymm8_signed = std::is_same<ScalarType, qasymm8_signed_t>::value;
156 svbool_t pg = wrapper::svwhilelt<ScalarType>(x, input_width);
157 svbool_t pg_0 = svunpklo(svunpklo(pg));
158 svbool_t pg_1 = svunpkhi(svunpklo(pg));
159 svbool_t pg_2 = svunpklo(svunpkhi(pg));
160 svbool_t pg_3 = svunpkhi(svunpkhi(pg));
163 auto vec_in_0 = svld1_f32(pg_0, tmp_ptr + x);
164 auto vec_in_1 = svld1_f32(pg_1, tmp_ptr + x + inc_1);
165 auto vec_in_2 = svld1_f32(pg_2, tmp_ptr + x + inc_2);
166 auto vec_in_3 = svld1_f32(pg_3, tmp_ptr + x + inc_3);
175 res_0 = svsub_f32_z(pg_0, vec_in_0, svdup_n_f32(sum));
176 res_1 = svsub_f32_z(pg_1, vec_in_1, svdup_n_f32(sum));
177 res_2 = svsub_f32_z(pg_2, vec_in_2, svdup_n_f32(sum));
178 res_3 = svsub_f32_z(pg_3, vec_in_3, svdup_n_f32(sum));
182 res_0 = svmul_f32_z(pg_0, vec_in_0, svdup_n_f32(sum));
183 res_1 = svmul_f32_z(pg_1, vec_in_1, svdup_n_f32(sum));
184 res_2 = svmul_f32_z(pg_2, vec_in_2, svdup_n_f32(sum));
185 res_3 = svmul_f32_z(pg_3, vec_in_3, svdup_n_f32(sum));
187 if (is_qasymm8_signed)
189 const auto offset_vec = svdup_n_f32(128.f);
190 res_0 = svsub_z(pg_0, res_0, offset_vec);
191 res_1 = svsub_z(pg_1, res_1, offset_vec);
192 res_2 = svsub_z(pg_2, res_2, offset_vec);
193 res_3 = svsub_z(pg_3, res_3, offset_vec);
198 const auto out = convert_float_to_int<SVEType>(res_0, res_1, res_2, res_3);
199 svst1(pg, out_ptr + x, out);
200 x += wrapper::svcnt<ScalarType>();
201 pg = wrapper::svwhilelt<ScalarType>(x, input_width);
202 pg_0 = svunpklo(svunpklo(pg));
203 pg_1 = svunpkhi(svunpklo(pg));
204 pg_2 = svunpklo(svunpkhi(pg));
205 pg_3 = svunpkhi(svunpkhi(pg));
206 }
while (svptest_any(all_true_pg, pg));
209 in_it, max_it, out_it);