reference_rmsnorm2d_default_epilogue Struct Reference

reference_rmsnorm2d_default_epilogue Struct Reference#

Composable Kernel: ck_tile::reference_rmsnorm2d_default_epilogue Struct Reference
Public Member Functions | List of all members
ck_tile::reference_rmsnorm2d_default_epilogue Struct Reference

#include <reference_rmsnorm2d_fwd.hpp>

Public Member Functions

template<typename OutDataType, typename AccDataType>
void operator() (int m, HostTensor< OutDataType > &o, const HostTensor< AccDataType > &acc)
template<typename OutDataType, typename AccDataType>
auto operator() (int m, const HostTensor< AccDataType > &acc)

Member Function Documentation

◆ operator()() [1/2]

template<typename OutDataType, typename AccDataType>
auto ck_tile::reference_rmsnorm2d_default_epilogue::operator() ( int m,
const HostTensor< AccDataType > & acc )
inline

◆ operator()() [2/2]

template<typename OutDataType, typename AccDataType>
void ck_tile::reference_rmsnorm2d_default_epilogue::operator() ( int m,
HostTensor< OutDataType > & o,
const HostTensor< AccDataType > & acc )
inline
The documentation for this struct was generated from the following file: