Searched refs:bitwise_binary_op (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/executorch/kernels/optimized/vec/ |
| H A D | vec_base.h | 529 static inline Vectorized<T> bitwise_binary_op(const Vectorized<T> &a, const Vectorized<T> &b, Op op… in bitwise_binary_op() function
553 return bitwise_binary_op(a, b, [](int_vector a, int_vector b) { return _mm256_and_si256(a, b); });
555 return bitwise_binary_op(a, b, [](int_vector a, int_vector b) { return _mm512_and_si512(a, b); });
562 return bitwise_binary_op(a, b, [](int_vector a, int_vector b) { return _mm256_or_si256(a, b); });
564 return bitwise_binary_op(a, b, [](int_vector a, int_vector b) { return _mm512_or_si512(a, b); });
571 return bitwise_binary_op(a, b, [](int_vector a, int_vector b) { return _mm256_xor_si256(a, b); });
573 return bitwise_binary_op(a, b, [](int_vector a, int_vector b) { return _mm512_xor_si512(a, b); });
587 static inline Vectorized<T> bitwise_binary_op(const Vectorized<T> &a, const Vectorized<T> &b, Op op… in bitwise_binary_op() function
611 return bitwise_binary_op(a, b, std::bit_and<intmax_t>());
615 return bitwise_binary_op(a, b, std::bit_or<intmax_t>());
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cpu/vec/ |
| H A D | vec_base.h | 771 static inline Vectorized<T> bitwise_binary_op(const Vectorized<T> &a, const Vectorized<T> &b, Op op… in bitwise_binary_op() function
795 return bitwise_binary_op(a, b, [](int_vector a, int_vector b) { return _mm256_and_si256(a, b); });
797 return bitwise_binary_op(a, b, [](int_vector a, int_vector b) { return _mm512_and_si512(a, b); });
804 return bitwise_binary_op(a, b, [](int_vector a, int_vector b) { return _mm256_or_si256(a, b); });
806 return bitwise_binary_op(a, b, [](int_vector a, int_vector b) { return _mm512_or_si512(a, b); });
813 return bitwise_binary_op(a, b, [](int_vector a, int_vector b) { return _mm256_xor_si256(a, b); });
815 return bitwise_binary_op(a, b, [](int_vector a, int_vector b) { return _mm512_xor_si512(a, b); });
829 static inline Vectorized<T> bitwise_binary_op(const Vectorized<T> &a, const Vectorized<T> &b, Op op… in bitwise_binary_op() function
853 return bitwise_binary_op(a, b, std::bit_and<intmax_t>());
857 return bitwise_binary_op(a, b, std::bit_or<intmax_t>());
[all …]
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | eval.cpp | 208 InterpValue bitwise_binary_op( in bitwise_binary_op() function in torch::jit::tensorexpr::SimpleIREvaluatorImpl
320 value_ = bitwise_binary_op<Type>(lhs_v, rhs_v, expr_type); \ in visit_binary_op()
325 value_ = bitwise_binary_op<unsigned char>(lhs_v, rhs_v, expr_type); in visit_binary_op()
|