Home
last modified time | relevance | path

Searched refs:bessel_i0 (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/special_math/
Dspecial_math_op_bessel.cc20 REGISTER3(UnaryOp, CPU, "BesselI0", functor::bessel_i0, Eigen::half, float,
50 REGISTER3(UnaryOp, GPU, "BesselI0", functor::bessel_i0, Eigen::half, float,
Dspecial_math_op_gpu_bessel.cu.cc23 DEFINE_UNARY3(bessel_i0, Eigen::half, float, double);
Dspecial_math_op_misc_impl.h691 struct bessel_i0 : base<T, Eigen::internal::scalar_bessel_i0_op<T>> {}; struct
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.math.special.pbtxt4 name: "bessel_i0"
Dtensorflow.math.pbtxt64 name: "bessel_i0"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.math.special.pbtxt4 name: "bessel_i0"
Dtensorflow.math.pbtxt64 name: "bessel_i0"
/external/tensorflow/tensorflow/python/ops/
Dspecial_math_ops_test.py411 self.assertAllClose(1., special_math_ops.bessel_i0(0.))
415 self.assertTrue(np.isnan(self.evaluate(special_math_ops.bessel_i0(np.nan))))
446 self.evaluate(special_math_ops.bessel_i0(x)),
447 self.evaluate(special_math_ops.bessel_i0(-x)))
478 special.i0(x), self.evaluate(special_math_ops.bessel_i0(x)))
583 special_math_ops.bessel_i0, inputs)
Dspecial_math_ops.py255 def bessel_i0(x, name=None): function
278 return gen_special_math_ops.bessel_i0(x)
Dmath_grad.py954 special_math_ops.bessel_i0(x) - math_ops.div(y, x))
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops.td1237 `bessel_i0e(x) = exp(-abs(x)) bessel_i0(x)`.
1239 This function is faster and numerically stabler than `bessel_i0(x)`.
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py2798 @RegisterPForWithArgs("BesselI0", special_math_ops.bessel_i0)