Member List

• ck
• GridwiseWelfordSecondHalfLayernorm2d

ck::GridwiseWelfordSecondHalfLayernorm2d< EMeanVarDataType, HDataType, GammaDataType, BetaDataType, ComputeDataType, EHGridDesc_M_N, MeanVarGridDesc_M_NBlock, CountGridDesc_M_NBlock, GammaBetaGridDesc_N, HElementwiseOperation, BlockSize, MThreadClusterSize, NThreadClusterSize, MThreadSliceSize, NThreadSliceSize, ESrcVectorSize, HDstVectorSize, GammaSrcVectorSize, BetaSrcVectorSize > Member List

This is the complete list of members for ck::GridwiseWelfordSecondHalfLayernorm2d< EMeanVarDataType, HDataType, GammaDataType, BetaDataType, ComputeDataType, EHGridDesc_M_N, MeanVarGridDesc_M_NBlock, CountGridDesc_M_NBlock, GammaBetaGridDesc_N, HElementwiseOperation, BlockSize, MThreadClusterSize, NThreadClusterSize, MThreadSliceSize, NThreadSliceSize, ESrcVectorSize, HDstVectorSize, GammaSrcVectorSize, BetaSrcVectorSize >, including all inherited members.

BlockwiseWelford typedef	ck::GridwiseWelfordSecondHalfLayernorm2d< EMeanVarDataType, HDataType, GammaDataType, BetaDataType, ComputeDataType, EHGridDesc_M_N, MeanVarGridDesc_M_NBlock, CountGridDesc_M_NBlock, GammaBetaGridDesc_N, HElementwiseOperation, BlockSize, MThreadClusterSize, NThreadClusterSize, MThreadSliceSize, NThreadSliceSize, ESrcVectorSize, HDstVectorSize, GammaSrcVectorSize, BetaSrcVectorSize >
I0	ck::GridwiseWelfordSecondHalfLayernorm2d< EMeanVarDataType, HDataType, GammaDataType, BetaDataType, ComputeDataType, EHGridDesc_M_N, MeanVarGridDesc_M_NBlock, CountGridDesc_M_NBlock, GammaBetaGridDesc_N, HElementwiseOperation, BlockSize, MThreadClusterSize, NThreadClusterSize, MThreadSliceSize, NThreadSliceSize, ESrcVectorSize, HDstVectorSize, GammaSrcVectorSize, BetaSrcVectorSize >	static
I1	ck::GridwiseWelfordSecondHalfLayernorm2d< EMeanVarDataType, HDataType, GammaDataType, BetaDataType, ComputeDataType, EHGridDesc_M_N, MeanVarGridDesc_M_NBlock, CountGridDesc_M_NBlock, GammaBetaGridDesc_N, HElementwiseOperation, BlockSize, MThreadClusterSize, NThreadClusterSize, MThreadSliceSize, NThreadSliceSize, ESrcVectorSize, HDstVectorSize, GammaSrcVectorSize, BetaSrcVectorSize >	static
M_BlockTileSize	ck::GridwiseWelfordSecondHalfLayernorm2d< EMeanVarDataType, HDataType, GammaDataType, BetaDataType, ComputeDataType, EHGridDesc_M_N, MeanVarGridDesc_M_NBlock, CountGridDesc_M_NBlock, GammaBetaGridDesc_N, HElementwiseOperation, BlockSize, MThreadClusterSize, NThreadClusterSize, MThreadSliceSize, NThreadSliceSize, ESrcVectorSize, HDstVectorSize, GammaSrcVectorSize, BetaSrcVectorSize >	static
N_BlockTileSize	ck::GridwiseWelfordSecondHalfLayernorm2d< EMeanVarDataType, HDataType, GammaDataType, BetaDataType, ComputeDataType, EHGridDesc_M_N, MeanVarGridDesc_M_NBlock, CountGridDesc_M_NBlock, GammaBetaGridDesc_N, HElementwiseOperation, BlockSize, MThreadClusterSize, NThreadClusterSize, MThreadSliceSize, NThreadSliceSize, ESrcVectorSize, HDstVectorSize, GammaSrcVectorSize, BetaSrcVectorSize >	static
Run(const EMeanVarDataType *__restrict__ p_e_grid, const EMeanVarDataType *__restrict__ p_in_welford_mean_grid, const EMeanVarDataType *__restrict__ p_in_welford_var_grid, const int32_t *__restrict__ p_in_welford_count_grid, const GammaDataType *__restrict__ p_gamma_grid, const BetaDataType *__restrict__ p_beta_grid, HDataType *__restrict__ p_h_grid, const EHGridDesc_M_N &e_grid_desc_m_n, const EHGridDesc_M_N &h_grid_desc_m_n, const MeanVarGridDesc_M_NBlock &mean_var_grid_desc_m_nblock, const CountGridDesc_M_NBlock &count_grid_desc_m_nblock, const GammaBetaGridDesc_N &gamma_grid_desc_n, const GammaBetaGridDesc_N &beta_grid_desc_n, index_t numMeanVarCountBlockTileIteration_N, index_t NBlockClusterLength, ComputeDataType epsilon, HElementwiseOperation h_element_op)	ck::GridwiseWelfordSecondHalfLayernorm2d< EMeanVarDataType, HDataType, GammaDataType, BetaDataType, ComputeDataType, EHGridDesc_M_N, MeanVarGridDesc_M_NBlock, CountGridDesc_M_NBlock, GammaBetaGridDesc_N, HElementwiseOperation, BlockSize, MThreadClusterSize, NThreadClusterSize, MThreadSliceSize, NThreadSliceSize, ESrcVectorSize, HDstVectorSize, GammaSrcVectorSize, BetaSrcVectorSize >	inlinestatic
thread_buffer_desc_m_1	ck::GridwiseWelfordSecondHalfLayernorm2d< EMeanVarDataType, HDataType, GammaDataType, BetaDataType, ComputeDataType, EHGridDesc_M_N, MeanVarGridDesc_M_NBlock, CountGridDesc_M_NBlock, GammaBetaGridDesc_N, HElementwiseOperation, BlockSize, MThreadClusterSize, NThreadClusterSize, MThreadSliceSize, NThreadSliceSize, ESrcVectorSize, HDstVectorSize, GammaSrcVectorSize, BetaSrcVectorSize >	static
thread_buffer_desc_m_n	ck::GridwiseWelfordSecondHalfLayernorm2d< EMeanVarDataType, HDataType, GammaDataType, BetaDataType, ComputeDataType, EHGridDesc_M_N, MeanVarGridDesc_M_NBlock, CountGridDesc_M_NBlock, GammaBetaGridDesc_N, HElementwiseOperation, BlockSize, MThreadClusterSize, NThreadClusterSize, MThreadSliceSize, NThreadSliceSize, ESrcVectorSize, HDstVectorSize, GammaSrcVectorSize, BetaSrcVectorSize >	static
thread_buffer_desc_n	ck::GridwiseWelfordSecondHalfLayernorm2d< EMeanVarDataType, HDataType, GammaDataType, BetaDataType, ComputeDataType, EHGridDesc_M_N, MeanVarGridDesc_M_NBlock, CountGridDesc_M_NBlock, GammaBetaGridDesc_N, HElementwiseOperation, BlockSize, MThreadClusterSize, NThreadClusterSize, MThreadSliceSize, NThreadSliceSize, ESrcVectorSize, HDstVectorSize, GammaSrcVectorSize, BetaSrcVectorSize >	static
thread_cluster_desc_m_n	ck::GridwiseWelfordSecondHalfLayernorm2d< EMeanVarDataType, HDataType, GammaDataType, BetaDataType, ComputeDataType, EHGridDesc_M_N, MeanVarGridDesc_M_NBlock, CountGridDesc_M_NBlock, GammaBetaGridDesc_N, HElementwiseOperation, BlockSize, MThreadClusterSize, NThreadClusterSize, MThreadSliceSize, NThreadSliceSize, ESrcVectorSize, HDstVectorSize, GammaSrcVectorSize, BetaSrcVectorSize >	static
ThreadBufferDimAccessOrder typedef	ck::GridwiseWelfordSecondHalfLayernorm2d< EMeanVarDataType, HDataType, GammaDataType, BetaDataType, ComputeDataType, EHGridDesc_M_N, MeanVarGridDesc_M_NBlock, CountGridDesc_M_NBlock, GammaBetaGridDesc_N, HElementwiseOperation, BlockSize, MThreadClusterSize, NThreadClusterSize, MThreadSliceSize, NThreadSliceSize, ESrcVectorSize, HDstVectorSize, GammaSrcVectorSize, BetaSrcVectorSize >
ThreadBufferLengths_M_1 typedef	ck::GridwiseWelfordSecondHalfLayernorm2d< EMeanVarDataType, HDataType, GammaDataType, BetaDataType, ComputeDataType, EHGridDesc_M_N, MeanVarGridDesc_M_NBlock, CountGridDesc_M_NBlock, GammaBetaGridDesc_N, HElementwiseOperation, BlockSize, MThreadClusterSize, NThreadClusterSize, MThreadSliceSize, NThreadSliceSize, ESrcVectorSize, HDstVectorSize, GammaSrcVectorSize, BetaSrcVectorSize >
ThreadBufferLengths_M_N typedef	ck::GridwiseWelfordSecondHalfLayernorm2d< EMeanVarDataType, HDataType, GammaDataType, BetaDataType, ComputeDataType, EHGridDesc_M_N, MeanVarGridDesc_M_NBlock, CountGridDesc_M_NBlock, GammaBetaGridDesc_N, HElementwiseOperation, BlockSize, MThreadClusterSize, NThreadClusterSize, MThreadSliceSize, NThreadSliceSize, ESrcVectorSize, HDstVectorSize, GammaSrcVectorSize, BetaSrcVectorSize >
ThreadBufferLengths_N typedef	ck::GridwiseWelfordSecondHalfLayernorm2d< EMeanVarDataType, HDataType, GammaDataType, BetaDataType, ComputeDataType, EHGridDesc_M_N, MeanVarGridDesc_M_NBlock, CountGridDesc_M_NBlock, GammaBetaGridDesc_N, HElementwiseOperation, BlockSize, MThreadClusterSize, NThreadClusterSize, MThreadSliceSize, NThreadSliceSize, ESrcVectorSize, HDstVectorSize, GammaSrcVectorSize, BetaSrcVectorSize >
ThreadClusterArrangeOrder typedef	ck::GridwiseWelfordSecondHalfLayernorm2d< EMeanVarDataType, HDataType, GammaDataType, BetaDataType, ComputeDataType, EHGridDesc_M_N, MeanVarGridDesc_M_NBlock, CountGridDesc_M_NBlock, GammaBetaGridDesc_N, HElementwiseOperation, BlockSize, MThreadClusterSize, NThreadClusterSize, MThreadSliceSize, NThreadSliceSize, ESrcVectorSize, HDstVectorSize, GammaSrcVectorSize, BetaSrcVectorSize >
ThreadClusterLengths_M_N typedef	ck::GridwiseWelfordSecondHalfLayernorm2d< EMeanVarDataType, HDataType, GammaDataType, BetaDataType, ComputeDataType, EHGridDesc_M_N, MeanVarGridDesc_M_NBlock, CountGridDesc_M_NBlock, GammaBetaGridDesc_N, HElementwiseOperation, BlockSize, MThreadClusterSize, NThreadClusterSize, MThreadSliceSize, NThreadSliceSize, ESrcVectorSize, HDstVectorSize, GammaSrcVectorSize, BetaSrcVectorSize >
ThreadWelfordDstDesc_M typedef	ck::GridwiseWelfordSecondHalfLayernorm2d< EMeanVarDataType, HDataType, GammaDataType, BetaDataType, ComputeDataType, EHGridDesc_M_N, MeanVarGridDesc_M_NBlock, CountGridDesc_M_NBlock, GammaBetaGridDesc_N, HElementwiseOperation, BlockSize, MThreadClusterSize, NThreadClusterSize, MThreadSliceSize, NThreadSliceSize, ESrcVectorSize, HDstVectorSize, GammaSrcVectorSize, BetaSrcVectorSize >
ThreadWelfordSrcDesc_M_1 typedef	ck::GridwiseWelfordSecondHalfLayernorm2d< EMeanVarDataType, HDataType, GammaDataType, BetaDataType, ComputeDataType, EHGridDesc_M_N, MeanVarGridDesc_M_NBlock, CountGridDesc_M_NBlock, GammaBetaGridDesc_N, HElementwiseOperation, BlockSize, MThreadClusterSize, NThreadClusterSize, MThreadSliceSize, NThreadSliceSize, ESrcVectorSize, HDstVectorSize, GammaSrcVectorSize, BetaSrcVectorSize >
ThreadwiseWelford typedef	ck::GridwiseWelfordSecondHalfLayernorm2d< EMeanVarDataType, HDataType, GammaDataType, BetaDataType, ComputeDataType, EHGridDesc_M_N, MeanVarGridDesc_M_NBlock, CountGridDesc_M_NBlock, GammaBetaGridDesc_N, HElementwiseOperation, BlockSize, MThreadClusterSize, NThreadClusterSize, MThreadSliceSize, NThreadSliceSize, ESrcVectorSize, HDstVectorSize, GammaSrcVectorSize, BetaSrcVectorSize >