ScaleAddScaleAddRelu Struct Reference

ScaleAddScaleAddRelu Struct Reference#

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

#include <element_wise_operation.hpp>

Public Member Functions

 ScaleAddScaleAddRelu (const float alpha1=1.f, const float alpha2=1.f)
template<typename E, typename C, typename D0, typename D1>
__host__ __device__ constexpr void operator() (E &e, const C &c, const D0 &d0, const D1 &d1) const
template<>
__host__ __device__ constexpr void operator() (float &e, const float &c, const float &d0, const float &d1) const
template<>
__host__ __device__ constexpr void operator() (half_t &e, const half_t &c, const half_t &d0, const half_t &d1) const
template<>
__host__ __device__ constexpr void operator() (bhalf_t &e, const bhalf_t &c, const bhalf_t &d0, const bhalf_t &d1) const
template<>
__host__ __device__ constexpr void operator() (int8_t &e, const int8_t &c, const float &d0, const float &d1) const

Public Attributes

const float alpha1_
const float alpha2_

Static Public Attributes

static constexpr const char * name = "ScaleAddScaleAddRelu"

Constructor & Destructor Documentation

◆ ScaleAddScaleAddRelu()

ck::tensor_operation::element_wise::ScaleAddScaleAddRelu::ScaleAddScaleAddRelu ( const float alpha1 = 1.f,
const float alpha2 = 1.f )
inline

Member Function Documentation

◆ operator()() [1/5]

template<>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::ScaleAddScaleAddRelu::operator() ( bhalf_t & e,
const bhalf_t & c,
const bhalf_t & d0,
const bhalf_t & d1 ) const
inlineconstexpr

◆ operator()() [2/5]

template<typename E, typename C, typename D0, typename D1>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::ScaleAddScaleAddRelu::operator() ( E & e,
const C & c,
const D0 & d0,
const D1 & d1 ) const
constexpr

◆ operator()() [3/5]

template<>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::ScaleAddScaleAddRelu::operator() ( float & e,
const float & c,
const float & d0,
const float & d1 ) const
inlineconstexpr

◆ operator()() [4/5]

template<>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::ScaleAddScaleAddRelu::operator() ( half_t & e,
const half_t & c,
const half_t & d0,
const half_t & d1 ) const
inlineconstexpr

◆ operator()() [5/5]

template<>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::ScaleAddScaleAddRelu::operator() ( int8_t & e,
const int8_t & c,
const float & d0,
const float & d1 ) const
inlineconstexpr

Member Data Documentation

◆ alpha1_

const float ck::tensor_operation::element_wise::ScaleAddScaleAddRelu::alpha1_

◆ alpha2_

const float ck::tensor_operation::element_wise::ScaleAddScaleAddRelu::alpha2_

◆ name

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