Gelu Struct Reference

Gelu Struct Reference#

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

#include <unary_element_wise_operation.hpp>

Public Member Functions

template<typename Y, typename X>
__host__ __device__ void operator() (Y &y, const X &x) const
template<>
__host__ __device__ void operator() (float &y, const float &x) const
template<>
__host__ __device__ void operator() (half_t &y, const half_t &x) const

Static Public Attributes

static constexpr const char * name = "Gelu"

Member Function Documentation

◆ operator()() [1/3]

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

◆ operator()() [2/3]

template<>
__host__ __device__ void ck::tensor_operation::element_wise::Gelu::operator() ( half_t & y,
const half_t & x ) const
inline

◆ operator()() [3/3]

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

Member Data Documentation

◆ name

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