diff options
author | 2018-06-12 02:33:46 -0700 | |
---|---|---|
committer | 2018-06-12 02:36:43 -0700 | |
commit | 433ac81400c788557001789f0a0c5a76a9b7e29c (patch) | |
tree | ead02fc17f972eeb48de06a4488754f2975530d3 /tensorflow/core/ops/math_ops.cc | |
parent | da88bfa02f6fb7071a41ff065ec9a918b1e0b1d6 (diff) |
Modified Bessel functions of order zero and one.
The functions are tf.math.bessel_i0(x), tf.math.bessel_i0e(x), tf.math.bessel_i1(x) and tf.math.bessel_i1e(x). The exponentially scaled versions tf.math.bessel_i0e(x) and tf.math.bessel_i1e(x) are more numerically stable. This code wraps the implementation that was recently added to Eigen.
PiperOrigin-RevId: 200186968
Diffstat (limited to 'tensorflow/core/ops/math_ops.cc')
-rw-r--r-- | tensorflow/core/ops/math_ops.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/core/ops/math_ops.cc b/tensorflow/core/ops/math_ops.cc index 8c0b073ce4..1740fa152c 100644 --- a/tensorflow/core/ops/math_ops.cc +++ b/tensorflow/core/ops/math_ops.cc @@ -239,6 +239,10 @@ REGISTER_OP("Acos").UNARY(); REGISTER_OP("Atan").UNARY(); +REGISTER_OP("BesselI0e").UNARY_REAL(); + +REGISTER_OP("BesselI1e").UNARY_REAL(); + #undef UNARY #undef UNARY_REAL #undef UNARY_COMPLEX |