Searched refs:__half (Results 1 – 4 of 4) sorted by relevance
23 using Eigen::half_impl::__half; in test_conversion()40 float val1 = float(half(__half(0x3c00))); in test_conversion()41 float val2 = float(half(__half(0x3c01))); in test_conversion()42 float val3 = float(half(__half(0x3c02))); in test_conversion()58 VERIFY_IS_EQUAL(float(half(__half(0x0000))), 0.0f); in test_conversion()59 VERIFY_IS_EQUAL(float(half(__half(0x3c00))), 1.0f); in test_conversion()62 VERIFY_IS_APPROX(float(half(__half(0x8001))), -5.96046e-08f); in test_conversion()63 VERIFY_IS_APPROX(float(half(__half(0x0001))), 5.96046e-08f); in test_conversion()64 VERIFY_IS_APPROX(float(half(__half(0x0002))), 1.19209e-07f); in test_conversion()69 VERIFY((numext::isinf)(float(half(__half(0xfc00))))); in test_conversion()[all …]
55 struct __half { struct56 EIGEN_DEVICE_FUNC __half() {} in __half() argument57 explicit EIGEN_DEVICE_FUNC __half(unsigned short raw) : x(raw) {} in __half() argument63 EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC __half raw_uint16_to_half(unsigned short x); argument64 EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC __half float_to_half_rtne(float ff);65 EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC float half_to_float(__half h);67 struct half_base : public __half {69 EIGEN_DEVICE_FUNC half_base(const half_base& h) : __half(h) {} in half_base()70 EIGEN_DEVICE_FUNC half_base(const __half& h) : __half(h) {} in half_base()78 typedef half_impl::__half __half; typedef[all …]
109 __half a1 = __low2half(kernel.packet[0]);110 __half a2 = __high2half(kernel.packet[0]);111 __half b1 = __low2half(kernel.packet[1]);112 __half b2 = __high2half(kernel.packet[1]);211 __half r1 = a1 < b1 ? __low2half(a) : __low2half(b);212 __half r2 = a2 < b2 ? __high2half(a) : __high2half(b);221 __half r1 = a1 > b1 ? __low2half(a) : __low2half(b);222 __half r2 = a2 > b2 ? __high2half(a) : __high2half(b);238 __half first = __low2half(a);239 __half second = __high2half(a);[all …]
2396 reinterpret_cast<const __half *>(GpuMemory(a) + batch * stride_a); in DoBlasGemmStridedBatched()2398 reinterpret_cast<const __half *>(GpuMemory(b) + batch * stride_b); in DoBlasGemmStridedBatched()2400 reinterpret_cast<__half *>(GpuMemoryMutable(c) + batch * stride_c); in DoBlasGemmStridedBatched()