Swish Struct Reference

Swish Struct Reference#

Composable Kernel: ck_tile::element_wise::Swish Struct Reference
Public Member Functions | Public Attributes | Static Public Attributes | List of all members
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_

Static Public Attributes

static constexpr const char * name = "Swish"

Constructor & Destructor Documentation

◆ Swish()

ck_tile::element_wise::Swish::Swish ( float beta = 1.0f)
inline

Member Function Documentation

◆ operator()()

template<typename Y, typename X>
CK_TILE_HOST_DEVICE void ck_tile::element_wise::Swish::operator() ( Y & y,
const X & x ) const
inline

Member Data Documentation

◆ beta_

const float ck_tile::element_wise::Swish::beta_

◆ name

const char* ck_tile::element_wise::Swish::name = "Swish"
staticconstexpr
The documentation for this struct was generated from the following file: