| /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));
|
