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