Swish Struct Reference#
ck_tile::element_wise::Swish Struct Reference
#include <unary_element_wise_operation.hpp>
Public Member Functions | |
Swish (float beta=1.0f) | |
template<typename Y , typename X > | |
CK_TILE_HOST_DEVICE void | operator() (Y &y, const X &x) const |
Public Attributes | |
const float | beta_ |
Constructor & Destructor Documentation
◆ Swish()
|
inline |
Member Function Documentation
◆ operator()()
template<typename Y , typename X >
|
inline |
Member Data Documentation
◆ beta_
const float ck_tile::element_wise::Swish::beta_ |
The documentation for this struct was generated from the following file:
- /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/elementwise/unary_element_wise_operation.hpp