diff options
| author |  Dan Ringwalt <ringwalt@google.com> | 2017-05-05 09:09:05 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-05-05 10:26:00 -0700 |
| commit | 692fad20f913ffa2cb874a87578ecabb03cc4557 (patch) | |
| tree | 172717f537c91b0d1ac0366731b4eb2093fb743b /tensorflow/core/kernels/cwise_ops.h | |
| parent | b329dd821e29e64c93b1b9bf38e61871c6cb53da (diff) | |
Merge changes from github.
Change: 155209832
Diffstat (limited to 'tensorflow/core/kernels/cwise_ops.h')
| -rw-r--r-- | tensorflow/core/kernels/cwise_ops.h | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/cwise_ops.h b/tensorflow/core/kernels/cwise_ops.h index fb4c649a82..423307fd4c 100644 --- a/tensorflow/core/kernels/cwise_ops.h +++ b/tensorflow/core/kernels/cwise_ops.h @@ -658,6 +658,22 @@ struct zeta : base<T, Eigen::internal::scalar_zeta_op<T>> {}; template <typename T> struct polygamma : base<T, Eigen::internal::scalar_polygamma_op<T>> {}; +template <typename Scalar> +struct scalar_atan2_op { + EIGEN_EMPTY_STRUCT_CTOR(scalar_atan2_op) + EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Scalar + operator()(const Scalar& y, const Scalar& x) const { +#if GOOGLE_CUDA + return ::atan2(y, x); +#else + return std::atan2(y, x); +#endif + } +}; + +template <typename T> +struct atan2 : base<T, scalar_atan2_op<T>> {}; + template <typename T> struct squared_difference : base<T, Eigen::internal::scalar_compose_op< |