Home
last modified time | relevance | path

Searched refs:Packet16f (Results 1 – 6 of 6) sorted by relevance

/external/eigen/Eigen/src/Core/arch/AVX512/
DMathFunctions.h21 const Packet16f p16f_##NAME = pset1<Packet16f>(X)
24 const Packet16f p16f_##NAME = (__m512)pset1<Packet16i>(X)
38 EIGEN_DEFINE_FUNCTION_ALLOWING_MULTIPLE_DEFINITIONS EIGEN_UNUSED Packet16f
39 plog<Packet16f>(const Packet16f& _x) {
40 Packet16f x = _x;
76 Packet16f emm0 = _mm512_cvtepi32_ps(_mm512_srli_epi32((__m512i)x, 23));
77 Packet16f e = _mm512_sub_ps(emm0, p16f_126f);
91 Packet16f tmp = _mm512_mask_blend_ps(mask, x, _mm512_setzero_ps());
96 Packet16f x2 = pmul(x, x);
97 Packet16f x3 = pmul(x2, x);
[all …]
DPacketMath.h31 typedef __m512 Packet16f; typedef
50 typedef Packet16f type;
98 struct unpacket_traits<Packet16f> {
117 EIGEN_STRONG_INLINE Packet16f pset1<Packet16f>(const float& from) {
130 EIGEN_STRONG_INLINE Packet16f pload1<Packet16f>(const float* from) {
139 EIGEN_STRONG_INLINE Packet16f plset<Packet16f>(const float& a) {
152 EIGEN_STRONG_INLINE Packet16f padd<Packet16f>(const Packet16f& a,
153 const Packet16f& b) {
163 EIGEN_STRONG_INLINE Packet16f psub<Packet16f>(const Packet16f& a,
164 const Packet16f& b) {
[all …]
/external/tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/src/FixedPoint/
DTypeCastingAVX512.h7 typedef __m512 Packet16f; typedef
16 EIGEN_STRONG_INLINE Packet16f pcast<Packet16q32i>(const Packet16q32i& a) {
26 EIGEN_STRONG_INLINE Packet16q32i pcast<Packet16f>(const Packet16f& a) {
36 EIGEN_STRONG_INLINE Packet32q16i pcast<Packet16f>(const Packet16f& a,
37 const Packet16f& b) {
62 EIGEN_STRONG_INLINE Packet64q8i pcast<Packet16f>(const Packet16f& a,
63 const Packet16f& b,
64 const Packet16f& c,
65 const Packet16f& d) {
/external/tensorflow/tensorflow/core/kernels/
Dsparse_matmul_op.h253 EIGEN_STRONG_INLINE Packet16f
254 pbroadcast_first<Packet16f>(const Packet16f& a_in) {
259 EIGEN_STRONG_INLINE Packet16f
260 pbroadcast_second<Packet16f>(const Packet16f& a_in) {
265 EIGEN_STRONG_INLINE Packet16f
266 pbroadcast_third<Packet16f>(const Packet16f& a_in) {
271 EIGEN_STRONG_INLINE Packet16f
272 pbroadcast_fourth<Packet16f>(const Packet16f& a_in) {
362 EIGEN_STRONG_INLINE Packet16f pload4bf16<Packet16f>(const float* from) {
370 EIGEN_STRONG_INLINE Packet16f pload2bf16<Packet16f>(const float* from) {
[all …]
/external/eigen/Eigen/src/Core/arch/CUDA/
DTypeCasting.h113 template<> EIGEN_STRONG_INLINE Packet16f pcast<Packet16h, Packet16f>(const Packet16h& a) {
126 template<> EIGEN_STRONG_INLINE Packet16h pcast<Packet16f, Packet16h>(const Packet16f& a) {
DPacketMathHalf.h418 EIGEN_STRONG_INLINE Packet16f half2float(const Packet16h& a) {
446 EIGEN_STRONG_INLINE Packet16h float2half(const Packet16f& a) {
480 Packet16f af = half2float(a);
481 Packet16f bf = half2float(b);
482 Packet16f rf = padd(af, bf);
487 Packet16f af = half2float(a);
488 Packet16f bf = half2float(b);
489 Packet16f rf = pmul(af, bf);
494 Packet16f from_float = half2float(from);