Home
last modified time | relevance | path

Searched defs:nan_mask (Results 1 – 7 of 7) sorted by relevance

/external/pytorch/test/cpp/api/
Dsupport.h136 auto nan_mask = x != x; in assert_tensor_not_equal() local
/external/pytorch/aten/src/ATen/cpu/vec/vec256/
Dvec256_double.h124 const auto nan_mask = _mm256_cmp_pd(not_nan_mask, zero_vec, _CMP_EQ_OQ); in angle() local
Dvec256_float.h132 const auto nan_mask = _mm256_cmp_ps(not_nan_mask, zero_vec, _CMP_EQ_OQ); in angle() local
Dvec256_bfloat16.h332 const auto nan_mask = _mm256_cmp_ps(not_nan_mask, zero_vec, _CMP_EQ_OQ); in angle() local
/external/pytorch/aten/src/ATen/cpu/vec/vec512/
Dvec512_double.h132 const auto nan_mask = _mm512_cmp_pd_mask(_mm512_castsi512_pd(not_nan), in angle() local
Dvec512_float.h151 const auto nan_mask = _mm512_cmp_ps_mask(_mm512_castsi512_ps(not_nan_vec), in angle() local
Dvec512_bfloat16.h415 const auto nan_mask = _mm512_cmp_ps_mask(_mm512_castsi512_ps(non_nan_mask_vec), in angle() local