Searched refs:half_impl (Results 1 – 12 of 12) sorted by relevance
50 namespace half_impl {76 struct half : public half_impl::half_base {78 typedef half_impl::__half __half;83 EIGEN_DEVICE_FUNC half(const __half& h) : half_impl::half_base(h) {} in half()84 EIGEN_DEVICE_FUNC half(const half& h) : half_impl::half_base(h) {} in half()87 : half_impl::half_base(half_impl::raw_uint16_to_half(b ? 0x3c00 : 0)) {} in half()90 : half_impl::half_base(half_impl::float_to_half_rtne(static_cast<float>(val))) {} in half()92 : half_impl::half_base(half_impl::float_to_half_rtne(f)) {} in half()99 return static_cast<signed char>(half_impl::half_to_float(*this)); in EIGEN_EXPLICIT_CAST()102 return static_cast<unsigned char>(half_impl::half_to_float(*this)); in EIGEN_EXPLICIT_CAST()[all …]
232 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 …]
452 namespace half_impl = Eigen::half_impl; in CudaAtomicCasHelper() local460 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()
113 Eigen::half_impl::raw_uint16_to_half(
197 return Eigen::half_impl::float_to_half_rtne(c); in ConvertUToT()
169 return Eigen::half_impl::half_to_float(x);175 return Eigen::half_impl::float_to_half_rtne(x);
23 using Eigen::half_impl::__half; in test_conversion()
37 return Eigen::half_impl::float_to_half_rtne(x); in operator ()()
1206 *out_min = Eigen::half_impl::half_to_float(*result.first); in GetWeightRange()1207 *out_max = Eigen::half_impl::half_to_float(*result.second); in GetWeightRange()1489 dst[i] = Eigen::half_impl::float_to_half_rtne(src[i]); in ConvertFP32ToFP16()3673 Eigen::half_impl::half_to_float(cast_vals_array[j][i]); in ConvertFusedBatchNorm()3680 Eigen::half_impl::half_to_float(cast_vals_array[j][0]); in ConvertFusedBatchNorm()
182 EXPECT_FLOAT_EQ(Eigen::half_impl::half_to_float(lhs[i]), in ExpectArrayNear()183 Eigen::half_impl::half_to_float(rhs[i])); in ExpectArrayNear()
140 return Eigen::half(half_impl::raw_uint16_to_half(__ldg(&address->x))); in loadConstant()
1295 return Eigen::half_impl::raw_uint16_to_half( in BitcastBetweenNativeTypes()