/external/tensorflow/tensorflow/compiler/xla/service/ |
hlo_evaluator_typed_visitor_half.cc | 21 template class HloEvaluatorTypedVisitor<Eigen::half, float>;
|
/external/swiftshader/src/Common/ |
Half.hpp | 20 class half class in namespace:sw 23 half() = default; 24 explicit half(float f); 28 half &operator=(half h); 29 half &operator=(float f); 35 inline half shortAsHalf(short s) 39 half h; 56 void toRGB16F(half rgb[3]) const 61 rgb[0] = half(R * factor) [all...] |
/external/tensorflow/tensorflow/core/kernels/ |
conv_2d_gpu_half.cu.cc | 32 template struct ShuffleAndReverse<Eigen::GpuDevice, Eigen::half, 4, int>; 33 template struct ShuffleAndReverse<Eigen::GpuDevice, Eigen::half, 4, 36 template struct TransformDepth<Eigen::GpuDevice, Eigen::half, int>; 38 template struct SwapDimension1And2InTensor3<Eigen::GpuDevice, Eigen::half>; 41 template struct TransformFilter<Eigen::GpuDevice, Eigen::half, int, 4>; 42 template struct ReverseTransformFilter<Eigen::GpuDevice, Eigen::half, 4>; 43 template struct NHWCToNCHW<Eigen::GpuDevice, Eigen::half, 4>; 44 template struct NCHWToNHWC<Eigen::GpuDevice, Eigen::half, 4>; 45 template struct PadInput<Eigen::GpuDevice, Eigen::half, int, 4>; 48 template struct TransformFilter<Eigen::GpuDevice, Eigen::half, int, 5> [all...] |
depthwise_conv_op_gpu_half.cu.cc | 25 template struct LaunchDepthwiseConvOp<GpuDevice, Eigen::half>; 26 template struct LaunchDepthwiseConvBackpropInputOp<GpuDevice, Eigen::half>; 27 template struct LaunchDepthwiseConvBackpropFilterOp<GpuDevice, Eigen::half>;
|
cwise_op_bessel.cu.cc | 22 DEFINE_UNARY3(bessel_i0e, Eigen::half, float, double); 23 DEFINE_UNARY3(bessel_i1e, Eigen::half, float, double);
|
cwise_op_gpu_inverse.cu.cc | 23 DEFINE_UNARY4(inverse, Eigen::half, float, double, int64); 24 DEFINE_SIMPLE_BINARY3(inverse_grad, Eigen::half, float, double);
|
cwise_op_gpu_rsqrt.cu.cc | 23 DEFINE_UNARY3(rsqrt, Eigen::half, float, double); 24 DEFINE_SIMPLE_BINARY3(rsqrt_grad, Eigen::half, float, double);
|
cwise_op_gpu_sigmoid.cu.cc | 23 DEFINE_UNARY3(sigmoid, Eigen::half, float, double); 24 DEFINE_SIMPLE_BINARY3(sigmoid_grad, Eigen::half, float, double);
|
cwise_op_gpu_sqrt.cu.cc | 23 DEFINE_UNARY3(sqrt, Eigen::half, float, double); 24 DEFINE_SIMPLE_BINARY3(sqrt_grad, Eigen::half, float, double);
|
cwise_op_gpu_tanh.cu.cc | 23 DEFINE_UNARY3(tanh, Eigen::half, float, double); 24 DEFINE_SIMPLE_BINARY3(tanh_grad, Eigen::half, float, double);
|
conv_ops_fused_half.cc | 27 DECLARE_FUNCTOR_GPU_SPEC(Eigen::half);
|
cwise_op_bessel.cc | 19 REGISTER3(UnaryOp, CPU, "BesselI0e", functor::bessel_i0e, Eigen::half, float, 21 REGISTER3(UnaryOp, CPU, "BesselI1e", functor::bessel_i1e, Eigen::half, float, 24 REGISTER3(UnaryOp, GPU, "BesselI0e", functor::bessel_i0e, Eigen::half, float, 26 REGISTER3(UnaryOp, GPU, "BesselI1e", functor::bessel_i1e, Eigen::half, float,
|
cwise_op_gpu_ceil.cu.cc | 22 DEFINE_UNARY3(ceil, Eigen::half, float, double);
|
cwise_op_gpu_cos.cu.cc | 22 DEFINE_UNARY3(cos, Eigen::half, float, double);
|
cwise_op_gpu_digamma.cu.cc | 22 DEFINE_UNARY3(digamma, Eigen::half, float, double);
|
cwise_op_gpu_erf.cu.cc | 22 DEFINE_UNARY3(erf, Eigen::half, float, double);
|
cwise_op_gpu_erfc.cu.cc | 22 DEFINE_UNARY3(erfc, Eigen::half, float, double);
|
cwise_op_gpu_expm1.cu.cc | 22 DEFINE_UNARY3(expm1, Eigen::half, float, double);
|
cwise_op_gpu_floor.cu.cc | 22 DEFINE_UNARY3(floor, Eigen::half, float, double);
|
cwise_op_gpu_isfinite.cu.cc | 22 DEFINE_UNARY3(isfinite, Eigen::half, float, double);
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
half_test.cc | 43 std::function<half(half)> compute_func; 51 std::vector<half> x({half(1.4), half(-2.3), half(3.2), half(-4.1), half(9.0), 52 half(42.0), half(-9.0), half(-100.0)}) [all...] |
/external/eigen/Eigen/src/Core/arch/CUDA/ |
Half.h | 29 // type Eigen::half (inheriting from CUDA's __half struct) with 48 struct half; 76 struct half : public half_impl::half_base { struct in namespace:Eigen 81 EIGEN_DEVICE_FUNC half() {} function in struct:Eigen::half 83 EIGEN_DEVICE_FUNC half(const __half& h) : half_impl::half_base(h) {} function in struct:Eigen::half 84 EIGEN_DEVICE_FUNC half(const half& h) : half_impl::half_base(h) {} function in struct:Eigen::half 86 explicit EIGEN_DEVICE_FUNC half(bool b) function in struct:Eigen::half 89 explicit EIGEN_DEVICE_FUNC half(const T& val) function in struct:Eigen::half 91 explicit EIGEN_DEVICE_FUNC half(float f function in struct:Eigen::half [all...] |
/external/eigen/test/ |
half_float.cpp | 12 #include <Eigen/src/Core/arch/CUDA/Half.h> 14 // Make sure it's possible to forward declare Eigen::half 16 struct half; 19 using Eigen::half; 26 VERIFY_IS_EQUAL(half(1.0f).x, 0x3c00); 27 VERIFY_IS_EQUAL(half(0.5f).x, 0x3800); 28 VERIFY_IS_EQUAL(half(0.33333f).x, 0x3555); 29 VERIFY_IS_EQUAL(half(0.0f).x, 0x0000); 30 VERIFY_IS_EQUAL(half(-0.0f).x, 0x8000); 31 VERIFY_IS_EQUAL(half(65504.0f).x, 0x7bff) [all...] |
/external/eigen/unsupported/Eigen/src/SpecialFunctions/ |
SpecialFunctionsHalf.h | 15 template<> EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Eigen::half lgamma(const Eigen::half& a) { 16 return Eigen::half(Eigen::numext::lgamma(static_cast<float>(a))); 18 template<> EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Eigen::half digamma(const Eigen::half& a) { 19 return Eigen::half(Eigen::numext::digamma(static_cast<float>(a))); 21 template<> EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Eigen::half zeta(const Eigen::half& x, const Eigen::half& q) { 22 return Eigen::half(Eigen::numext::zeta(static_cast<float>(x), static_cast<float>(q))) [all...] |
/external/tensorflow/tensorflow/core/lib/bfloat16/ |
bfloat16.cc | 25 B16_DEVICE_FUNC bfloat16::operator Eigen::half() const { 26 return static_cast<Eigen::half>(float(*this));
|