|
| template<typename Y , typename X0 , typename X1 > |
| __host__ constexpr __device__ void | operator() (Y &y, const X0 &x0, const X1 &x1) const |
| |
| template<> |
| __host__ constexpr __device__ void | operator() (float &y, const float &x0, const float &x1) const |
| |
| template<> |
| __host__ constexpr __device__ void | operator() (double &y, const double &x0, const double &x1) const |
| |
| template<> |
| __host__ constexpr __device__ void | operator() (float &y, const float &x0, const half_t &x1) const |
| |
| template<> |
| __host__ constexpr __device__ void | operator() (half_t &y, const float &x0, const float &x1) const |
| |
| template<> |
| __host__ constexpr __device__ void | operator() (half_t &y, const float &x0, const half_t &x1) const |
| |
| template<> |
| __host__ constexpr __device__ void | operator() (half_t &y, const half_t &x0, const half_t &x1) const |
| |
| template<> |
| __host__ constexpr __device__ void | operator() (float &y, const float &x0, const bf16_t &x1) const |
| |
| template<> |
| __host__ constexpr __device__ void | operator() (bf16_t &y, const bf16_t &x0, const bf16_t &x1) const |
| |
| template<> |
| __host__ constexpr __device__ void | operator() (bf16_t &y, const float &x0, const bf16_t &x1) const |
| |
| template<> |
| __host__ constexpr __device__ void | operator() (int8_t &y, const int8_t &x0, const int8_t &x1) const |
| |