Home
last modified time | relevance | path

Searched refs:vec_elements_flt (Results 1 – 1 of 1) sorted by relevance

/external/ComputeLibrary/src/cpu/kernels/softmax/generic/neon/
Dimpl.cpp140 auto vec_elements_flt = convert_int_to_float<float32x4x4_t>(vec_elements); in neon_softmax_logits_1d_quantized() local
144 vec_elements_flt.val[0] = vmulq_f32(vec_elements_flt.val[0], scale_beta_vec); in neon_softmax_logits_1d_quantized()
145 vec_elements_flt.val[1] = vmulq_f32(vec_elements_flt.val[1], scale_beta_vec); in neon_softmax_logits_1d_quantized()
146 vec_elements_flt.val[2] = vmulq_f32(vec_elements_flt.val[2], scale_beta_vec); in neon_softmax_logits_1d_quantized()
147 vec_elements_flt.val[3] = vmulq_f32(vec_elements_flt.val[3], scale_beta_vec); in neon_softmax_logits_1d_quantized()
148 … vec_sum.val[0] = vaddq_f32(vec_sum.val[0], vexpq_f32(vec_elements_flt.val[0])); in neon_softmax_logits_1d_quantized()
149 … vec_sum.val[1] = vaddq_f32(vec_sum.val[1], vexpq_f32(vec_elements_flt.val[1])); in neon_softmax_logits_1d_quantized()
150 … vec_sum.val[2] = vaddq_f32(vec_sum.val[2], vexpq_f32(vec_elements_flt.val[2])); in neon_softmax_logits_1d_quantized()
151 … vec_sum.val[3] = vaddq_f32(vec_sum.val[3], vexpq_f32(vec_elements_flt.val[3])); in neon_softmax_logits_1d_quantized()
155vec_elements_flt.val[0] = vexpq_f32(vmulq_f32(vec_elements_flt.val[0], scale_beta_vec)); in neon_softmax_logits_1d_quantized()
[all …]