Home
last modified time | relevance | path

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

/cmsis-nn-3.6.0-3.5.0/Source/SoftmaxFunctions/
Darm_nn_softmax_common_s8.c93 const int32_t shifted_scale = ONE_OVER1((sum > 0 ? sum << headroom : 0) - (1 << 31)); in arm_nn_softmax_common_s8() local
109 … DIV_POW2(MUL_SAT(shifted_scale, EXP_ON_NEG(MUL_SAT(diff * mask, mult))), bits_over_unit) + in arm_nn_softmax_common_s8()
131 … DIV_POW2(MUL_SAT(shifted_scale, EXP_ON_NEG(MUL_SAT(diff * mask, mult))), bits_over_unit) + in arm_nn_softmax_common_s8()
Darm_softmax_u8.c81 const int32_t shifted_scale = ONE_OVER1((sum << headroom) - (1 << 31)); in arm_softmax_u8() local
89 … DIV_POW2(MUL_SAT(shifted_scale, EXP_ON_NEG(MUL_SAT(diff * mask, mult))), bits_over_unit); in arm_softmax_u8()
Darm_softmax_s8.c152 const int32_t shifted_scale = ONE_OVER1((sum > 0 ? sum << headroom : 0) - (1 << 31)); in arm_softmax_s8() local
172 tmp_res = MUL_SAT_MVE(vdupq_n_s32(shifted_scale), tmp_res); in arm_softmax_s8()
195 … DIV_POW2(MUL_SAT(shifted_scale, EXP_ON_NEG(MUL_SAT(diff * mask, mult))), bits_over_unit) + in arm_softmax_s8()