Searched defs:nan_mask (Results 1 – 7 of 7) sorted by relevance
136 auto nan_mask = x != x; in assert_tensor_not_equal() local
124 const auto nan_mask = _mm256_cmp_pd(not_nan_mask, zero_vec, _CMP_EQ_OQ); in angle() local
132 const auto nan_mask = _mm256_cmp_ps(not_nan_mask, zero_vec, _CMP_EQ_OQ); in angle() local
332 const auto nan_mask = _mm256_cmp_ps(not_nan_mask, zero_vec, _CMP_EQ_OQ); in angle() local
132 const auto nan_mask = _mm512_cmp_pd_mask(_mm512_castsi512_pd(not_nan), in angle() local
151 const auto nan_mask = _mm512_cmp_ps_mask(_mm512_castsi512_ps(not_nan_vec), in angle() local
415 const auto nan_mask = _mm512_cmp_ps_mask(_mm512_castsi512_ps(non_nan_mask_vec), in angle() local