LeakyRelu Struct Reference

• ck_tile
• element_wise
• LeakyRelu

Public Member Functions | Public Attributes | Static Public Attributes | List of all members

ck_tile::element_wise::LeakyRelu Struct Reference

#include <unary_element_wise_operation.hpp>

Public Member Functions
	LeakyRelu (float alpha=0.01f)
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 = "LeakyRelu"

Constructor & Destructor Documentation

◆ LeakyRelu()

ck_tile::element_wise::LeakyRelu::LeakyRelu ( float alpha = 0.01f )

inline

Member Function Documentation

◆ operator()()

template<typename T>

CK_TILE_HOST_DEVICE void ck_tile::element_wise::LeakyRelu::operator() ( T & y, const T & x ) const

inline

Member Data Documentation

◆ alpha_

const float ck_tile::element_wise::LeakyRelu::alpha_

◆ name

const char* ck_tile::element_wise::LeakyRelu::name = "LeakyRelu"

staticconstexpr

---

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

• tile/ops/elementwise/unary_element_wise_operation.hpp