Searched refs:mask_num_plus_nudge_overflow (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
H A D | avx2_quantization_utils.h | 56 const __m256i mask_num_plus_nudge_overflow = _mm256_cmpgt_epi32( in rounding_right_shift() local 62 mask_num_plus_nudge_overflow); in rounding_right_shift() 78 const __m256i mask_num_plus_nudge_overflow = _mm256_cmpgt_epi32( in rounding_right_shift() local 84 mask_num_plus_nudge_overflow); in rounding_right_shift()
|
/aosp_15_r20/external/ruy/ruy/ |
H A D | kernel_avx2_fma.cc | 485 const __m256i mask_num_plus_nudge_overflow = _mm256_cmpgt_epi32( 490 shifted_sum, one_shift_31minus_exp, mask_num_plus_nudge_overflow); 960 const __m256i mask_num_plus_nudge_overflow = _mm256_cmpgt_epi32( 965 shifted_sum, one_shift_31minus_exp, mask_num_plus_nudge_overflow);
|
H A D | kernel_avx512.cc | 378 const auto mask_num_plus_nudge_overflow = _mm512_cmpgt_epi64_mask( 383 shifted_sum, mask_num_plus_nudge_overflow, one_shift_31minus_exp); 797 const auto mask_num_plus_nudge_overflow = _mm512_cmpgt_epi64_mask( 802 shifted_sum, mask_num_plus_nudge_overflow, one_shift_31minus_exp);
|
H A D | kernel_avx.cc | 859 const __m256i mask_num_plus_nudge_overflow = 866 shifted_sum, one_shift_31minus_exp, mask_num_plus_nudge_overflow); 1402 const __m256i mask_num_plus_nudge_overflow = 1409 shifted_sum, one_shift_31minus_exp, mask_num_plus_nudge_overflow);
|