Home
last modified time | relevance | path

Searched refs:bessel_i0e (Results 1 – 25 of 31) sorted by relevance

12

/external/eigen/unsupported/Eigen/src/SpecialFunctions/arch/GPU/
DSpecialFunctions.h209 using numext::bessel_i0e;
210 return make_float4(bessel_i0e(x.x), bessel_i0e(x.y), bessel_i0e(x.z), bessel_i0e(x.w));
216 using numext::bessel_i0e;
217 return make_double2(bessel_i0e(x.x), bessel_i0e(x.y));
/external/tensorflow/tensorflow/core/kernels/special_math/
Dspecial_math_op_bessel.cc24 REGISTER3(UnaryOp, CPU, "BesselI0e", functor::bessel_i0e, Eigen::half, float,
54 REGISTER3(UnaryOp, GPU, "BesselI0e", functor::bessel_i0e, Eigen::half, float,
Dspecial_math_op_gpu_bessel.cu.cc25 DEFINE_UNARY3(bessel_i0e, Eigen::half, float, double);
Dspecial_math_op_misc_impl.h689 struct bessel_i0e : base<T, Eigen::internal::scalar_bessel_i0e_op<T>> {}; struct
/external/eigen/unsupported/Eigen/src/SpecialFunctions/
DBesselFunctionsHalf.h20 EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Eigen::half bessel_i0e(const Eigen::half& x) { in bessel_i0e() function
21 return Eigen::half(Eigen::numext::bessel_i0e(static_cast<float>(x))); in bessel_i0e()
DBesselFunctionsBFloat16.h20 EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Eigen::bfloat16 bessel_i0e(const Eigen::bfloat16& x) { in bessel_i0e() function
21 return Eigen::bfloat16(Eigen::numext::bessel_i0e(static_cast<float>(x))); in bessel_i0e()
DBesselFunctionsFunctors.h55 using numext::bessel_i0e;
56 return bessel_i0e(x);
DHipVectorCompatibility.h43 HIP_SCALAR_ACCESSOR_BUILDER_RETVAL(bessel_i0e)
DBesselFunctionsPacketMath.h30 return numext::bessel_i0e(x); in pbessel_i0e()
DBesselFunctionsArrayAPI.h55 bessel_i0e(const Eigen::ArrayBase<Derived>& x) { in bessel_i0e() function
DBesselFunctionsImpl.h1890 EIGEN_DEVICE_FUNC inline EIGEN_MATHFUNC_RETVAL(bessel_i0e, Scalar)
1891 bessel_i0e(const Scalar& x) {
1892 return EIGEN_MATHFUNC_IMPL(bessel_i0e, Scalar)::run(x);
/external/tensorflow/tensorflow/python/ops/signal/
Dwindow_ops.py88 special_math_ops.bessel_i0e(num) / special_math_ops.bessel_i0e(beta))
/external/tensorflow/tensorflow/python/ops/
Dspecial_math_ops_test.py407 self.assertAllClose(1., special_math_ops.bessel_i0e(0.))
412 np.isnan(self.evaluate(special_math_ops.bessel_i0e(np.nan))))
445 self.evaluate(special_math_ops.bessel_i0e(x)),
446 self.evaluate(special_math_ops.bessel_i0e(-x)))
477 special.i0e(x), self.evaluate(special_math_ops.bessel_i0e(x)))
529 special.i0e(x), self.evaluate(special_math_ops.bessel_i0e(x)))
582 special_math_ops.bessel_i0e, inputs)
Dspecial_math_ops.py284 def bessel_i0e(x, name=None): function
305 return gen_special_math_ops.bessel_i0e(x)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.math.special.pbtxt8 name: "bessel_i0e"
Dtensorflow.math.pbtxt76 name: "bessel_i0e"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.math.special.pbtxt8 name: "bessel_i0e"
/external/eigen/unsupported/test/
Dbessel_functions.cpp66 CALL_SUBTEST(res = bessel_i0e(x); in array_bessel_functions()
Dspecial_packetmath.cpp62 CHECK_CWISE1_IF(PacketTraits::HasBessel, numext::bessel_i0e, internal::pbessel_i0e); in packetmath_real()
/external/eigen/unsupported/Eigen/
DSpecialFunctions43 * - bessel_i0e
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor_test_ops.py91 special_math_ops.bessel_i0e,
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py108 bessel_i0e = _unary_op(special_math_ops.bessel_i0e) variable
/external/tensorflow/tensorflow/python/kernel_tests/math_ops/
Dcwise_ops_unary_test.py226 self._compareBoth(x, special.i0e, special_math_ops.bessel_i0e)
281 self._compareBoth(x, special.i0e, special_math_ops.bessel_i0e)
335 self._compareBoth(x, special.i0e, special_math_ops.bessel_i0e)
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-with-tf2xla.mlir247 // CHECK-LABEL: bessel_i0e
248 func.func @bessel_i0e(%arg0: tensor<3xf16>, %arg1: tensor<3xf32>, %arg2: tensor<3xf64>) -> (tensor<…
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dmath_test.py105 special_math_ops.bessel_i0e,

12