Function marian::gpu::RMSNormalizationGrad

Function Documentation

void marian::gpu::RMSNormalizationGrad(Ptr<Allocator> allocator, Tensor gradX, Tensor gradGamma, Tensor gradBeta, Tensor adj, Tensor y, Tensor x, Tensor gamma, Tensor beta, float eps)