AddReluAdd Struct Reference

AddReluAdd Struct Reference#

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

#include <element_wise_operation.hpp>

Public Member Functions

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

Static Public Attributes

static constexpr const char * name = "AddReluAdd"

Member Function Documentation

◆ operator()() [1/7]

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

◆ operator()() [2/7]

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

◆ operator()() [3/7]

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

◆ operator()() [4/7]

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

◆ operator()() [5/7]

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

◆ operator()() [6/7]

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

◆ operator()() [7/7]

template<typename Y, typename X0, typename X1, typename X2>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::AddReluAdd::operator() ( Y & ,
const X0 & ,
const X1 & ,
const X2 &  ) const
constexpr

Member Data Documentation

◆ name

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