SoftRelu Struct Reference

SoftRelu Struct Reference#

Composable Kernel: ck_tile::element_wise::SoftRelu Struct Reference
ck_tile::element_wise::SoftRelu Struct Reference

#include <unary_element_wise_operation.hpp>

Public Member Functions

 SoftRelu (float alpha=1.f)
template<typename T>
CK_TILE_HOST_DEVICE void operator() (T &y, const T &x) const

Public Attributes

const float alpha_

Static Public Attributes

static constexpr const char * name = "SoftRelu"

Constructor & Destructor Documentation

◆ SoftRelu()

ck_tile::element_wise::SoftRelu::SoftRelu ( float alpha = 1.f)
inline

Member Function Documentation

◆ operator()()

template<typename T>
CK_TILE_HOST_DEVICE void ck_tile::element_wise::SoftRelu::operator() ( T & y,
const T & x ) const
inline

Member Data Documentation

◆ alpha_

const float ck_tile::element_wise::SoftRelu::alpha_

◆ name

const char* ck_tile::element_wise::SoftRelu::name = "SoftRelu"
staticconstexpr

The documentation for this struct was generated from the following file: