Relu Struct Reference

Relu Struct Reference#

Composable Kernel: ck_tile::element_wise::Relu Struct Reference
Public Member Functions | Static Public Attributes | List of all members
ck_tile::element_wise::Relu Struct Reference

#include <unary_element_wise_operation.hpp>

Public Member Functions

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

Static Public Attributes

static constexpr const char * name = "Relu"

Member Function Documentation

◆ operator()() [1/2]

template<>
CK_TILE_HOST_DEVICE void ck_tile::element_wise::Relu::operator() ( ck_tile::bf16_t & y,
const ck_tile::bf16_t & x ) const
inline

◆ operator()() [2/2]

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

Member Data Documentation

◆ name

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