Home
last modified time | relevance | path

Searched refs:bessel_k0 (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/special_math/
Dspecial_math_op_bessel.cc29 REGISTER3(UnaryOp, CPU, "BesselK0", functor::bessel_k0, Eigen::half, float,
59 REGISTER3(UnaryOp, GPU, "BesselK0", functor::bessel_k0, Eigen::half, float,
Dspecial_math_op_gpu_bessel.cu.cc28 DEFINE_UNARY3(bessel_k0, Eigen::half, float, double);
Dspecial_math_op_misc_impl.h703 struct bessel_k0 : base<T, Eigen::internal::scalar_bessel_k0_op<T>> {}; struct
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.math.special.pbtxt28 name: "bessel_k0"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.math.special.pbtxt28 name: "bessel_k0"
/external/tensorflow/tensorflow/python/ops/
Dspecial_math_ops_test.py431 self.assertTrue(np.isinf(self.evaluate(special_math_ops.bessel_k0(0.))))
435 self.assertTrue(np.isnan(self.evaluate(special_math_ops.bessel_k0(np.nan))))
506 special.k0(x), self.evaluate(special_math_ops.bessel_k0(x)))
558 special.k0(x), self.evaluate(special_math_ops.bessel_k0(x)))
611 special_math_ops.bessel_k0, inputs)
Dspecial_math_ops.py363 def bessel_k0(x, name=None): function
386 return gen_special_math_ops.bessel_k0(x)
Dmath_grad.py1003 partial_x = -special_math_ops.bessel_k0(x) - math_ops.div(y, x)
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py2802 @RegisterPForWithArgs("BesselK0", special_math_ops.bessel_k0)