ConvScaleAdd Struct Reference

ConvScaleAdd Struct Reference#

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

#include <binary_element_wise_operation.hpp>

Public Member Functions

__host__ __device__ ConvScaleAdd (float scale_in=1.f, float scale_wei=1.f, float scale_out=1.f)
template<typename E, typename C, typename D>
__host__ __device__ void operator() (E &e, const C &c, const D &d) const
template<>
__host__ __device__ void operator() (f8_t &e, const float &c, const float &d) const

Public Attributes

float scale_in_
float scale_wei_
float scale_out_

Static Public Attributes

static constexpr const char * name = "ConvScaleAdd"

Constructor & Destructor Documentation

◆ ConvScaleAdd()

__host__ __device__ ck::tensor_operation::element_wise::ConvScaleAdd::ConvScaleAdd ( float scale_in = 1.f,
float scale_wei = 1.f,
float scale_out = 1.f )
inline

Member Function Documentation

◆ operator()() [1/2]

template<typename E, typename C, typename D>
__host__ __device__ void ck::tensor_operation::element_wise::ConvScaleAdd::operator() ( E & e,
const C & c,
const D & d ) const

◆ operator()() [2/2]

template<>
__host__ __device__ void ck::tensor_operation::element_wise::ConvScaleAdd::operator() ( f8_t & e,
const float & c,
const float & d ) const
inline

Member Data Documentation

◆ name

const char* ck::tensor_operation::element_wise::ConvScaleAdd::name = "ConvScaleAdd"
staticconstexpr

◆ scale_in_

float ck::tensor_operation::element_wise::ConvScaleAdd::scale_in_

◆ scale_out_

float ck::tensor_operation::element_wise::ConvScaleAdd::scale_out_

◆ scale_wei_

float ck::tensor_operation::element_wise::ConvScaleAdd::scale_wei_
The documentation for this struct was generated from the following file: