| 1391 | /// \param arg function argument |
| 1392 | /// \return function value stored in single-preicision |
| 1393 | MEGDNN_HOST MEGDNN_DEVICE static expr atan(float arg) { |
| 1394 | #if defined(__CUDA_ARCH__) |
| 1395 | return expr(atanf(arg)); |
| 1396 | #else |
| 1397 | return expr(std::atan(arg)); |
| 1398 | #endif |
| 1399 | } |
| 1400 | |
| 1401 | /// Arc tangent implementation. |
| 1402 | /// \param x first argument |