#include <binary_element_wise_operation.hpp>
|
| template<typename T> |
| __host__ __device__ constexpr void | operator() (T &y, const T &x0, const T &x1) const |
| template<> |
| __host__ __device__ constexpr void | operator() (float &y, const float &x0, const float &x1) const |
| template<> |
| __host__ __device__ constexpr void | operator() (double &y, const double &x0, const double &x1) const |
| template<> |
| __host__ __device__ constexpr void | operator() (half_t &y, const half_t &x0, const half_t &x1) const |
|
| static constexpr const char * | name = "AddHardswish" |
◆ operator()() [1/4]
template<>
| __host__ __device__ constexpr void ck::tensor_operation::element_wise::AddHardswish::operator() |
( |
double & | y, |
|
|
const double & | x0, |
|
|
const double & | x1 ) const |
|
inlineconstexpr |
◆ operator()() [2/4]
template<>
| __host__ __device__ constexpr void ck::tensor_operation::element_wise::AddHardswish::operator() |
( |
float & | y, |
|
|
const float & | x0, |
|
|
const float & | x1 ) const |
|
inlineconstexpr |
◆ operator()() [3/4]
template<>
| __host__ __device__ constexpr void ck::tensor_operation::element_wise::AddHardswish::operator() |
( |
half_t & | y, |
|
|
const half_t & | x0, |
|
|
const half_t & | x1 ) const |
|
inlineconstexpr |
◆ operator()() [4/4]
template<typename T>
| __host__ __device__ constexpr void ck::tensor_operation::element_wise::AddHardswish::operator() |
( |
T & | y, |
|
|
const T & | x0, |
|
|
const T & | x1 ) const |
|
constexpr |
◆ name
| const char* ck::tensor_operation::element_wise::AddHardswish::name = "AddHardswish" |
|
staticconstexpr |
The documentation for this struct was generated from the following file: