Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/special_math/
Dspecial_math_op_bessel.cc31 REGISTER3(UnaryOp, CPU, "BesselK1", functor::bessel_k1, Eigen::half, float,
61 REGISTER3(UnaryOp, GPU, "BesselK1", functor::bessel_k1, Eigen::half, float,
Dspecial_math_op_gpu_bessel.cu.cc29 DEFINE_UNARY3(bessel_k1, Eigen::half, float, double);
Dspecial_math_op_misc_impl.h709 struct bessel_k1 : base<T, Eigen::internal::scalar_bessel_k1_op<T>> {}; struct
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.math.special.pbtxt36 name: "bessel_k1"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.math.special.pbtxt36 name: "bessel_k1"
/external/tensorflow/tensorflow/python/ops/
Dspecial_math_ops_test.py433 self.assertTrue(np.isinf(self.evaluate(special_math_ops.bessel_k1(0.))))
438 self.assertTrue(np.isnan(self.evaluate(special_math_ops.bessel_k1(np.nan))))
510 special.k1(x), self.evaluate(special_math_ops.bessel_k1(x)))
562 special.k1(x), self.evaluate(special_math_ops.bessel_k1(x)))
619 special_math_ops.bessel_k1, inputs)
Dspecial_math_ops.py417 def bessel_k1(x, name=None): function
440 return gen_special_math_ops.bessel_k1(x)
Dmath_grad.py981 partial_x = -special_math_ops.bessel_k1(x)
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py2803 @RegisterPForWithArgs("BesselK1", special_math_ops.bessel_k1)