ScaleAdd Struct Reference

• ck
• tensor_operation
• element_wise
• ScaleAdd

Public Member Functions | Public Attributes | Static Public Attributes | List of all members

ck::tensor_operation::element_wise::ScaleAdd Struct Reference

#include <binary_element_wise_operation.hpp>

Public Member Functions
__host__ __device__	ScaleAdd (float scale=1.f)
template<typename Y, typename X0, typename X1>
__host__ __device__ constexpr void	operator() (Y &y, const X0 &x0, const X1 &x1) const
template<>
__host__ __device__ void	operator() (float &y, const float &x0, const half_t &x1) const
template<>
__host__ __device__ void	operator() (float &y, const float &x0, const bhalf_t &x1) const

Public Attributes
float	scale_

Static Public Attributes
static constexpr const char *	name = "ScaleAdd"

Constructor & Destructor Documentation

◆ ScaleAdd()

__host__ __device__ ck::tensor_operation::element_wise::ScaleAdd::ScaleAdd ( float scale = 1.f )

inline

Member Function Documentation

◆ operator()() [1/3]

template<>

__host__ __device__ void ck::tensor_operation::element_wise::ScaleAdd::operator() ( float & y, const float & x0, const bhalf_t & x1 ) const

inline

◆ operator()() [2/3]

template<>

__host__ __device__ void ck::tensor_operation::element_wise::ScaleAdd::operator() ( float & y, const float & x0, const half_t & x1 ) const

inline

◆ operator()() [3/3]

template<typename Y, typename X0, typename X1>

__host__ __device__ constexpr void ck::tensor_operation::element_wise::ScaleAdd::operator() ( Y & y, const X0 & x0, const X1 & x1 ) const

inlineconstexpr

Member Data Documentation

◆ name

const char* ck::tensor_operation::element_wise::ScaleAdd::name = "ScaleAdd"

staticconstexpr

◆ scale_

float ck::tensor_operation::element_wise::ScaleAdd::scale_

---

The documentation for this struct was generated from the following file:

• binary_element_wise_operation.hpp