Swish Struct Reference

• ck
• tensor_operation
• element_wise
• Swish

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

ck::tensor_operation::element_wise::Swish Struct Reference

#include <unary_element_wise_operation.hpp>

Public Member Functions
	Swish (float beta=1.0f)
template<typename Y, typename X>
__host__ __device__ void	operator() (Y &y, const X &x) const
template<>
__host__ __device__ void	operator() (bhalf_t &y, const float &x) const

Public Attributes
const float	beta_

Static Public Attributes
static constexpr const char *	name = "Swish"

Constructor & Destructor Documentation

◆ Swish()

ck::tensor_operation::element_wise::Swish::Swish ( float beta = 1.0f )

inline

Member Function Documentation

◆ operator()() [1/2]

template<>

__host__ __device__ void ck::tensor_operation::element_wise::Swish::operator() ( bhalf_t & y, const float & x ) const

inline

◆ operator()() [2/2]

template<typename Y, typename X>

__host__ __device__ void ck::tensor_operation::element_wise::Swish::operator() ( Y & y, const X & x ) const

inline

Member Data Documentation

◆ beta_

const float ck::tensor_operation::element_wise::Swish::beta_

◆ name

const char* ck::tensor_operation::element_wise::Swish::name = "Swish"

staticconstexpr

---

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

• tensor_operation/gpu/element/unary_element_wise_operation.hpp