Composable Kernel: include/ck/tensor_operation/gpu/device/device_normalization_bwd_gamma_beta.hpp File Reference
#include <iostream>
#include <vector>
#include "ck/tensor_operation/gpu/device/device_base.hpp"
Go to the source code of this file.
|
template<typename DYDataType , typename XDataType , typename MeanInvStdDataType , typename DGammaDataType , typename DBetaDataType , index_t Rank, index_t NumReduceDim> |
using | ck::tensor_operation::device::DeviceNormalizationBwdGammaBetaPtr = std::unique_ptr< DeviceNormalizationBwdGammaBeta< DYDataType, XDataType, MeanInvStdDataType, DGammaDataType, DBetaDataType, Rank, NumReduceDim > > |
|