ConvScale Struct Reference

ConvScale Struct Reference#

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

#include <unary_element_wise_operation.hpp>

Public Member Functions

CK_TILE_HOST_DEVICE ConvScale (float scale_in=1.f, float scale_wei=1.f, float scale_out=1.f)
 
template<typename E , typename C >
CK_TILE_HOST_DEVICE void operator() (E &e, const C &c) const
 
template<>
CK_TILE_HOST_DEVICE void operator() (ck_tile::fp8_t &e, const float &c) const
 

Public Attributes

float scale_in_
 
float scale_wei_
 
float scale_out_
 

Constructor & Destructor Documentation

◆ ConvScale()

CK_TILE_HOST_DEVICE ck_tile::element_wise::ConvScale::ConvScale ( float  scale_in = 1.f,
float  scale_wei = 1.f,
float  scale_out = 1.f 
)
inline

Member Function Documentation

◆ operator()() [1/2]

template<>
CK_TILE_HOST_DEVICE void ck_tile::element_wise::ConvScale::operator() ( ck_tile::fp8_t &  e,
const float &  c 
) const
inline

◆ operator()() [2/2]

template<typename E , typename C >
CK_TILE_HOST_DEVICE void ck_tile::element_wise::ConvScale::operator() ( E &  e,
const C &  c 
) const

Member Data Documentation

◆ scale_in_

float ck_tile::element_wise::ConvScale::scale_in_

◆ scale_out_

float ck_tile::element_wise::ConvScale::scale_out_

◆ scale_wei_

float ck_tile::element_wise::ConvScale::scale_wei_
The documentation for this struct was generated from the following file:
  • /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck_tile/ops/elementwise/unary_element_wise_operation.hpp