Home
last modified time | relevance | path

Searched refs:Packet16i (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_matmul_op.h298 EIGEN_STRONG_INLINE Packet16i
299 pbroadcast_first<Packet16i>(const Packet16i& a_in) {
304 EIGEN_STRONG_INLINE Packet16i
305 pbroadcast_second<Packet16i>(const Packet16i& a_in) {
310 EIGEN_STRONG_INLINE Packet16i
311 pbroadcast_third<Packet16i>(const Packet16i& a_in) {
316 EIGEN_STRONG_INLINE Packet16i
317 pbroadcast_fourth<Packet16i>(const Packet16i& a_in) {
458 Packet16i tmp = _mm512_castps_si512(from); in pexpand_bf16_u()
459 Packet16i tmp2 = _mm512_alignr_epi32(tmp, tmp, 8); in pexpand_bf16_u()
/external/tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/src/FixedPoint/
DTypeCastingAVX512.h8 typedef __m512i Packet16i; typedef
38 Packet16i a_int = _mm512_cvtps_epi32(a);
39 Packet16i b_int = _mm512_cvtps_epi32(b);
66 Packet16i a_int = _mm512_cvtps_epi32(a);
67 Packet16i b_int = _mm512_cvtps_epi32(b);
68 Packet16i c_int = _mm512_cvtps_epi32(c);
69 Packet16i d_int = _mm512_cvtps_epi32(d);
/external/eigen/Eigen/src/Core/arch/AVX512/
DPacketMath.h32 typedef __m512i Packet16i; typedef
110 struct unpacket_traits<Packet16i> {
125 EIGEN_STRONG_INLINE Packet16i pset1<Packet16i>(const int& from) {
191 EIGEN_STRONG_INLINE Packet16i pconj(const Packet16i& a) {
441 EIGEN_STRONG_INLINE Packet16i pload<Packet16i>(const int* from) {
455 EIGEN_STRONG_INLINE Packet16i ploadu<Packet16i>(const int* from) {
546 EIGEN_STRONG_INLINE void pstore<int>(int* to, const Packet16i& from) {
560 EIGEN_STRONG_INLINE void pstoreu<int>(int* to, const Packet16i& from) {
568 Packet16i stride_vector = _mm512_set1_epi32(stride);
569 Packet16i stride_multiplier =
[all …]
DMathFunctions.h24 const Packet16f p16f_##NAME = (__m512)pset1<Packet16i>(X)
174 Packet16i emm0 = _mm512_cvttps_epi32(padd(m, p16f_127));