Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/special_math/
Dspecial_math_op_bessel.cc22 REGISTER3(UnaryOp, CPU, "BesselI1", functor::bessel_i1, Eigen::half, float,
52 REGISTER3(UnaryOp, GPU, "BesselI1", functor::bessel_i1, Eigen::half, float,
Dspecial_math_op_gpu_bessel.cu.cc24 DEFINE_UNARY3(bessel_i1, Eigen::half, float, double);
Dspecial_math_op_misc_impl.h697 struct bessel_i1 : base<T, Eigen::internal::scalar_bessel_i1_op<T>> {}; struct
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.math.special.pbtxt12 name: "bessel_i1"
Dtensorflow.math.pbtxt72 name: "bessel_i1"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.math.special.pbtxt12 name: "bessel_i1"
Dtensorflow.math.pbtxt72 name: "bessel_i1"
/external/tensorflow/tensorflow/python/ops/
Dspecial_math_ops_test.py413 self.assertAllClose(0., special_math_ops.bessel_i1(0.))
418 self.assertTrue(np.isnan(self.evaluate(special_math_ops.bessel_i1(np.nan))))
461 self.evaluate(special_math_ops.bessel_i1(x)),
462 self.evaluate(-special_math_ops.bessel_i1(-x)))
480 special.i1(x), self.evaluate(special_math_ops.bessel_i1(x)))
591 special_math_ops.bessel_i1, inputs)
Dspecial_math_ops.py309 def bessel_i1(x, name=None): function
332 return gen_special_math_ops.bessel_i1(x)
Dmath_grad.py927 partial_x = special_math_ops.bessel_i1(x)
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops.td1258 `bessel_i1e(x) = exp(-abs(x)) bessel_i1(x)`.
1260 This function is faster and numerically stabler than `bessel_i1(x)`.
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py2799 @RegisterPForWithArgs("BesselI1", special_math_ops.bessel_i1)