Home
last modified time | relevance | path

Searched refs:bessel_i1e (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/special_math/
Dspecial_math_op_bessel.cc26 REGISTER3(UnaryOp, CPU, "BesselI1e", functor::bessel_i1e, Eigen::half, float,
56 REGISTER3(UnaryOp, GPU, "BesselI1e", functor::bessel_i1e, Eigen::half, float,
Dspecial_math_op_gpu_bessel.cu.cc26 DEFINE_UNARY3(bessel_i1e, Eigen::half, float, double);
Dspecial_math_op_misc_impl.h700 struct bessel_i1e : base<T, Eigen::internal::scalar_bessel_i1e_op<T>> {}; struct
/external/tensorflow/tensorflow/python/ops/
Dspecial_math_ops_test.py414 self.assertAllClose(0., special_math_ops.bessel_i1e(0.))
420 np.isnan(self.evaluate(special_math_ops.bessel_i1e(np.nan))))
465 self.evaluate(special_math_ops.bessel_i1e(x)),
466 self.evaluate(-special_math_ops.bessel_i1e(-x)))
484 special.i1e(x), self.evaluate(special_math_ops.bessel_i1e(x)))
536 special.i1e(x), self.evaluate(special_math_ops.bessel_i1e(x)))
595 special_math_ops.bessel_i1e, inputs)
Dspecial_math_ops.py337 def bessel_i1e(x, name=None): function
358 return gen_special_math_ops.bessel_i1e(x)
Dsparse_ops.py3176 special_math_ops.bessel_i1e,
Dmath_grad.py937 partial_x = (special_math_ops.bessel_i1e(x) - math_ops.sign(x) * y)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.math.special.pbtxt16 name: "bessel_i1e"
Dtensorflow.math.pbtxt76 name: "bessel_i1e"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.math.special.pbtxt16 name: "bessel_i1e"
Dtensorflow.math.pbtxt76 name: "bessel_i1e"
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py109 bessel_i1e = _unary_op(special_math_ops.bessel_i1e) variable
/external/tensorflow/tensorflow/python/kernel_tests/
Dcwise_ops_unary_test.py230 self._compareBoth(x, special.i1e, special_math_ops.bessel_i1e)
283 self._compareBoth(x, special.i1e, special_math_ops.bessel_i1e)
337 self._compareBoth(x, special.i1e, special_math_ops.bessel_i1e)
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-with-tf2xla.mlir266 // CHECK-LABEL: bessel_i1e
267 func @bessel_i1e(%arg0: tensor<3xf16>, %arg1: tensor<3xf32>, %arg2: tensor<3xf64>) -> (tensor<3xf16…
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dmap_vectorization_test.py96 ("BesselI1e", special_math_ops.bessel_i1e),
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dmath_test.py114 special_math_ops.bessel_i1e,
Dpfor.py2801 @RegisterPForWithArgs("BesselI1e", special_math_ops.bessel_i1e)
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops.td1258 `bessel_i1e(x) = exp(-abs(x)) bessel_i1(x)`.