Home
last modified time | relevance | path

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

/cmsis-nn-3.7.0/Source/SoftmaxFunctions/
Darm_nn_softmax_common_s8.c80 int32_t diff = 0; in arm_nn_softmax_common_s8() local
85 diff = input[col] - max; in arm_nn_softmax_common_s8()
86 if (diff >= diff_min) in arm_nn_softmax_common_s8()
88 sum += DIV_POW2(EXP_ON_NEG(MUL_SAT(diff * mask, mult)), ACCUM_BITS); in arm_nn_softmax_common_s8()
104 diff = input[col] - max; in arm_nn_softmax_common_s8()
106 if (diff >= diff_min) in arm_nn_softmax_common_s8()
109 … DIV_POW2(MUL_SAT(shifted_scale, EXP_ON_NEG(MUL_SAT(diff * mask, mult))), bits_over_unit) + in arm_nn_softmax_common_s8()
127 diff = input[col] - max; in arm_nn_softmax_common_s8()
128 if (diff >= diff_min) 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.c67 int32_t diff = 0; in arm_softmax_u8() local
72 diff = input[col] - max; in arm_softmax_u8()
73 if (diff >= diff_min) in arm_softmax_u8()
75 sum += DIV_POW2(EXP_ON_NEG(MUL_SAT(diff * mask, mult)), ACCUM_BITS); in arm_softmax_u8()
85 diff = input[col] - max; in arm_softmax_u8()
86 if (diff >= diff_min) in arm_softmax_u8()
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.c143 const int32_t diff = input[tail_idx + i] - max; in arm_softmax_s8() local
144 if (diff >= diff_min) in arm_softmax_s8()
146 sum += DIV_POW2(EXP_ON_NEG(MUL_SAT(diff * mask, mult)), ACCUM_BITS); in arm_softmax_s8()
191 int32_t diff = input[tail_idx + i] - max; in arm_softmax_s8() local
192 if (diff >= diff_min) 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()
Darm_softmax_s16.c64 int32_t diff = 0; in arm_softmax_s16() local
70 diff = input[col] - max; in arm_softmax_s16()
71 const int32_t scaled_diff = arm_nn_requantize(diff, mult, shift); in arm_softmax_s16()
/cmsis-nn-3.7.0/
Dcheck_pdsc.sh34 diff $LISTED_SOURCE_FILES $ACTUAL_SOURCE_FILES
/cmsis-nn-3.7.0/Source/ConvolutionFunctions/
Darm_depthwise_conv_s8_opt.c262 const int diff = kernel_y - ker_y_end; in arm_depthwise_conv_s8_opt() local
263 if (diff != 0) in arm_depthwise_conv_s8_opt()
265 memset(&col_buffer[index], 0, (kernel_x * input_ch) * diff * sizeof(int16_t)); in arm_depthwise_conv_s8_opt()
Darm_depthwise_conv_fast_s16.c273 const int diff = kernel_y - ker_y_end; in arm_depthwise_conv_fast_s16() local
274 if (diff != 0) in arm_depthwise_conv_fast_s16()
276 memset(&col_buffer[index], 0, (kernel_x * input_ch) * diff * sizeof(int16_t)); in arm_depthwise_conv_fast_s16()
Darm_depthwise_conv_s4_opt.c304 const int diff = kernel_y - ker_y_end; in arm_depthwise_conv_s4_opt() local
305 if (diff != 0) in arm_depthwise_conv_s4_opt()
307 memset(&col_buffer[index], 0, (kernel_x * input_ch) * diff * sizeof(int16_t)); in arm_depthwise_conv_s4_opt()