gridwise_normalization_bwd_gamma_beta.hpp File Reference

gridwise_normalization_bwd_gamma_beta.hpp File Reference#

Composable Kernel: gridwise_normalization_bwd_gamma_beta.hpp File Reference
Classes | Namespaces
gridwise_normalization_bwd_gamma_beta.hpp File Reference
#include "ck/utility/data_type.hpp"
#include "ck/tensor_operation/gpu/thread/threadwise_tensor_slice_transfer.hpp"
#include "ck/tensor_operation/gpu/element/element_wise_operation.hpp"
#include "ck/tensor_operation/gpu/block/reduction_functions_blockwise.hpp"

Go to the source code of this file.

Classes

struct  ck::GridwiseNormalizationBwdGammaBeta_mk_to_k< DYDataType, XDataType, MeanInvStdDataType, ComputeDataType, DGammaDataType, DBetaDataType, GridDesc_M_K, GridDesc_M, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, DYSrcVectorDim, DYSrcVectorSize, XSrcVectorDim, XSrcVectorSize, MeanInvStdSrcVectorDim, MeanInvStdSrcVectorSize, DGammaDstVectorSize, DBetaDstVectorSize >

Namespaces

namespace  ck