/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/tensor_operation/gpu/device/device_normalization_bwd_data.hpp Source File#
device_normalization_bwd_data.hpp
Go to the documentation of this file.
std::unique_ptr< DeviceNormalizationBwdData< DYDataType, XDataType, GammaDataType, MeanInvStdDataType, DXDataType, Rank, NumReduceDim > > DeviceNormalizationBwdDataPtr
Definition: device_normalization_bwd_data.hpp:55
Definition: ck.hpp:267
Definition: device_base.hpp:77
Definition: device_normalization_bwd_data.hpp:22
virtual std::unique_ptr< BaseArgument > MakeArgumentPointer(const std::vector< index_t > lengths, const std::vector< index_t > dyStrides, const std::vector< index_t > xStrides, const std::vector< index_t > gammaStrides, const std::vector< index_t > meanStrides, const std::vector< index_t > invStdStrides, const std::vector< index_t > dxStrides, const std::vector< index_t > reduceDims, const void *p_dy, const void *p_x, const void *p_gamma, const void *p_mean, const void *p_invStd, void *p_dx)=0
virtual std::unique_ptr< BaseInvoker > MakeInvokerPointer()=0