| 1031 | struct scalar_atan2_op { |
| 1032 | EIGEN_EMPTY_STRUCT_CTOR(scalar_atan2_op) |
| 1033 | EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Scalar |
| 1034 | operator()(const Scalar& y, const Scalar& x) const { |
| 1035 | #if GOOGLE_CUDA |
| 1036 | return std::atan2(y, x); |
| 1037 | #elif TENSORFLOW_USE_ROCM |
| 1038 | return ::atan2(y, x); |
| 1039 | #else |
| 1040 | return std::atan2(y, x); |
| 1041 | #endif |
| 1042 | } |
| 1043 | }; |
| 1044 | |
| 1045 | template <typename T> |