Searched refs:diff (Results 1 – 6 of 6) sorted by relevance
| /cmsis-nn-3.6.0-3.5.0/Source/SoftmaxFunctions/ |
| D | arm_nn_softmax_common_s8.c | 80 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()
|
| D | arm_softmax_u8.c | 67 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()
|
| D | arm_softmax_s8.c | 143 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()
|
| D | arm_softmax_s16.c | 64 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.6.0-3.5.0/Source/ConvolutionFunctions/ |
| D | arm_depthwise_conv_s8_opt.c | 262 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()
|
| D | arm_depthwise_conv_fast_s16.c | 273 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()
|
