Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/special_math/
Dspecial_math_op_bessel.cc35 REGISTER3(UnaryOp, CPU, "BesselK1e", functor::bessel_k1e, Eigen::half, float,
65 REGISTER3(UnaryOp, GPU, "BesselK1e", functor::bessel_k1e, Eigen::half, float,
Dspecial_math_op_gpu_bessel.cu.cc31 DEFINE_UNARY3(bessel_k1e, Eigen::half, float, double);
Dspecial_math_op_misc_impl.h712 struct bessel_k1e : base<T, Eigen::internal::scalar_bessel_k1e_op<T>> {}; struct
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.math.special.pbtxt40 name: "bessel_k1e"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.math.special.pbtxt40 name: "bessel_k1e"
/external/tensorflow/tensorflow/python/ops/
Dspecial_math_ops_test.py434 self.assertTrue(np.isinf(self.evaluate(special_math_ops.bessel_k1e(0.))))
440 np.isnan(self.evaluate(special_math_ops.bessel_k1e(np.nan))))
512 special.k1e(x), self.evaluate(special_math_ops.bessel_k1e(x)))
564 special.k1e(x), self.evaluate(special_math_ops.bessel_k1e(x)))
623 special_math_ops.bessel_k1e, inputs)
Dspecial_math_ops.py445 def bessel_k1e(x, name=None): function
466 return gen_special_math_ops.bessel_k1e(x)
Dmath_grad.py991 partial_x = (y - special_math_ops.bessel_k1e(x))
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py2805 @RegisterPForWithArgs("BesselK1e", special_math_ops.bessel_k1e)