ClippedRelu Struct Reference

ClippedRelu Struct Reference#

Composable Kernel: ck::tensor_operation::element_wise::ClippedRelu Struct Reference
Public Member Functions | Public Attributes | Static Public Attributes | List of all members
ck::tensor_operation::element_wise::ClippedRelu Struct Reference

#include <unary_element_wise_operation.hpp>

Public Member Functions

 ClippedRelu (float alpha=0.f, float beta=1.f)
template<typename T>
__host__ __device__ void operator() (T &y, const T &x) const
template<typename Y, typename X>
__host__ __device__ constexpr void operator() (Y &y, const X &x) const
template<>
__host__ __device__ void operator() (bhalf_t &y, const float &x) const

Public Attributes

const float alpha_
const float beta_

Static Public Attributes

static constexpr const char * name = "ClippedRelu"

Constructor & Destructor Documentation

◆ ClippedRelu()

ck::tensor_operation::element_wise::ClippedRelu::ClippedRelu ( float alpha = 0.f,
float beta = 1.f )
inline

Member Function Documentation

◆ operator()() [1/3]

template<>
__host__ __device__ void ck::tensor_operation::element_wise::ClippedRelu::operator() ( bhalf_t & y,
const float & x ) const
inline

◆ operator()() [2/3]

template<typename T>
__host__ __device__ void ck::tensor_operation::element_wise::ClippedRelu::operator() ( T & y,
const T & x ) const
inline

◆ operator()() [3/3]

template<typename Y, typename X>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::ClippedRelu::operator() ( Y & y,
const X & x ) const
constexpr

Member Data Documentation

◆ alpha_

const float ck::tensor_operation::element_wise::ClippedRelu::alpha_

◆ beta_

const float ck::tensor_operation::element_wise::ClippedRelu::beta_

◆ name

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