Home
last modified time | relevance | path

Searched refs:raw_uint16_to_half (Results 1 – 7 of 7) sorted by relevance

/external/eigen/Eigen/src/Core/arch/CUDA/
DHalf.h63 EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC __half raw_uint16_to_half(unsigned short x);
87 : half_impl::half_base(half_impl::raw_uint16_to_half(b ? 0x3c00 : 0)) {} in half()
272 EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC __half raw_uint16_to_half(unsigned short x) { in raw_uint16_to_half() function
506 static Eigen::half (min)() { return Eigen::half_impl::raw_uint16_to_half(0x400); }
507 static Eigen::half lowest() { return Eigen::half_impl::raw_uint16_to_half(0xfbff); }
508 static Eigen::half (max)() { return Eigen::half_impl::raw_uint16_to_half(0x7bff); }
509 static Eigen::half epsilon() { return Eigen::half_impl::raw_uint16_to_half(0x0800); }
511 static Eigen::half infinity() { return Eigen::half_impl::raw_uint16_to_half(0x7c00); }
512 static Eigen::half quiet_NaN() { return Eigen::half_impl::raw_uint16_to_half(0x7e00); }
513 static Eigen::half signaling_NaN() { return Eigen::half_impl::raw_uint16_to_half(0x7e00); }
[all …]
DTypeCasting.h173 Eigen::half h = raw_uint16_to_half(static_cast<unsigned short>(a64));
175 h = raw_uint16_to_half(static_cast<unsigned short>(a64 >> 16));
177 h = raw_uint16_to_half(static_cast<unsigned short>(a64 >> 32));
179 h = raw_uint16_to_half(static_cast<unsigned short>(a64 >> 48));
DPacketMathHalf.h232 return Eigen::half(half_impl::raw_uint16_to_half(__float2half_rn(a1 + a2)));
266 return Eigen::half(half_impl::raw_uint16_to_half(__float2half_rn(a1 * a2)));
384 …return half_impl::raw_uint16_to_half(static_cast<unsigned short>(_mm256_extract_epi16(from.x, 0)));
752 return half_impl::raw_uint16_to_half(static_cast<unsigned short>(_mm_extract_epi16(from.x, 0)));
1011 return half_impl::raw_uint16_to_half(static_cast<unsigned short>(_mm_cvtsi64_si32(from.x)));
1022 Eigen::half ha = half_impl::raw_uint16_to_half(static_cast<unsigned short>(a64));
1023 Eigen::half hb = half_impl::raw_uint16_to_half(static_cast<unsigned short>(b64));
1025 ha = half_impl::raw_uint16_to_half(static_cast<unsigned short>(a64 >> 16));
1026 hb = half_impl::raw_uint16_to_half(static_cast<unsigned short>(b64 >> 16));
1028 ha = half_impl::raw_uint16_to_half(static_cast<unsigned short>(a64 >> 32));
[all …]
/external/tensorflow/tensorflow/core/util/
Dcuda_device_functions.h460 Eigen::half acc = accumulate(half_impl::raw_uint16_to_half(high)); in CudaAtomicCasHelper()
463 return half_impl::raw_uint16_to_half(static_cast<uint16>(result >> 16)); in CudaAtomicCasHelper()
469 Eigen::half acc = accumulate(half_impl::raw_uint16_to_half(low)); in CudaAtomicCasHelper()
472 return half_impl::raw_uint16_to_half(static_cast<uint16>(result & 0xffff)); in CudaAtomicCasHelper()
Dtensor_slice_util.h113 Eigen::half_impl::raw_uint16_to_half(
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorEvaluator.h140 return Eigen::half(half_impl::raw_uint16_to_half(__ldg(&address->x))); in loadConstant()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral.cc1295 return Eigen::half_impl::raw_uint16_to_half( in BitcastBetweenNativeTypes()