AddRelu Struct Reference

AddRelu Struct Reference#

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

#include <binary_element_wise_operation.hpp>

Public Member Functions

template<typename Y, typename X0, typename X1>
__host__ __device__ constexpr void operator() (Y &y, const X0 &x0, const X1 &x1) const
template<>
__host__ __device__ constexpr void operator() (float &y, const float &x0, const float &x1) const
template<>
__host__ __device__ constexpr void operator() (double &y, const double &x0, const double &x1) const
template<>
__host__ __device__ constexpr void operator() (half_t &y, const half_t &x0, const half_t &x1) const
template<>
__host__ __device__ constexpr void operator() (half_t &y, const float &x0, const half_t &x1) const
template<>
__host__ __device__ constexpr void operator() (float &y, const float &x0, const half_t &x1) const
template<>
__host__ __device__ constexpr void operator() (bhalf_t &y, const float &x0, const bhalf_t &x1) const
template<>
__host__ __device__ constexpr void operator() (bhalf_t &y, const bhalf_t &x0, const bhalf_t &x1) const
template<>
__host__ __device__ constexpr void operator() (int &y, const int &x0, const int8_t &x1) const
template<>
__host__ __device__ constexpr void operator() (int8_t &y, const int8_t &x0, const int8_t &x1) const

Static Public Attributes

static constexpr const char * name = "AddRelu"

Member Function Documentation

◆ operator()() [1/10]

template<>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::AddRelu::operator() ( bhalf_t & y,
const bhalf_t & x0,
const bhalf_t & x1 ) const
inlineconstexpr

◆ operator()() [2/10]

template<>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::AddRelu::operator() ( bhalf_t & y,
const float & x0,
const bhalf_t & x1 ) const
inlineconstexpr

◆ operator()() [3/10]

template<>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::AddRelu::operator() ( double & y,
const double & x0,
const double & x1 ) const
inlineconstexpr

◆ operator()() [4/10]

template<>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::AddRelu::operator() ( float & y,
const float & x0,
const float & x1 ) const
inlineconstexpr

◆ operator()() [5/10]

template<>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::AddRelu::operator() ( float & y,
const float & x0,
const half_t & x1 ) const
inlineconstexpr

◆ operator()() [6/10]

template<>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::AddRelu::operator() ( half_t & y,
const float & x0,
const half_t & x1 ) const
inlineconstexpr

◆ operator()() [7/10]

template<>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::AddRelu::operator() ( half_t & y,
const half_t & x0,
const half_t & x1 ) const
inlineconstexpr

◆ operator()() [8/10]

template<>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::AddRelu::operator() ( int & y,
const int & x0,
const int8_t & x1 ) const
inlineconstexpr

◆ operator()() [9/10]

template<>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::AddRelu::operator() ( int8_t & y,
const int8_t & x0,
const int8_t & x1 ) const
inlineconstexpr

◆ operator()() [10/10]

template<typename Y, typename X0, typename X1>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::AddRelu::operator() ( Y & y,
const X0 & x0,
const X1 & x1 ) const
constexpr

Member Data Documentation

◆ name

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