Searched refs:binary_compare (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/cpu/vec/vec256/ |
| D | vec256_bfloat16.h | 619 Vectorized<T> inline binary_compare(const Vectorized<T>& b, Op op) const { in binary_compare() function
631 … return binary_compare(other, [](__m256 x, __m256 y) { return _mm256_cmp_ps(x, y, _CMP_GT_OQ); });
634 … return binary_compare(other, [](__m256 x, __m256 y) { return _mm256_cmp_ps(x, y, _CMP_LT_OQ); });
637 … return binary_compare(other, [](__m256 x, __m256 y) { return _mm256_cmp_ps(x, y, _CMP_GE_OQ); });
640 … return binary_compare(other, [](__m256 x, __m256 y) { return _mm256_cmp_ps(x, y, _CMP_LE_OQ); });
643 … return binary_compare(other, [](__m256 x, __m256 y) { return _mm256_cmp_ps(x, y, _CMP_EQ_OQ); });
646 … return binary_compare(other, [](__m256 x, __m256 y) { return _mm256_cmp_ps(x, y, _CMP_NEQ_UQ); });
|
| /external/pytorch/aten/src/ATen/cpu/vec/vec512/ |
| D | vec512_bfloat16.h | 703 Vectorized<T> inline binary_compare(const Vectorized<T>& b, Op op) const { in binary_compare() function
715 return binary_compare(other, [](__m512 x, __m512 y) {
722 return binary_compare(other, [](__m512 x, __m512 y) {
729 return binary_compare(other, [](__m512 x, __m512 y) {
736 return binary_compare(other, [](__m512 x, __m512 y) {
743 return binary_compare(other, [](__m512 x, __m512 y) {
750 return binary_compare(other, [](__m512 x, __m512 y) {
|