Searched defs:part_grad_beta (Results 1 – 1 of 1) sorted by relevance
911 T_ACC* part_grad_beta) in cuComputePartGradGammaBeta()972 const T_ACC* part_grad_beta, in cuComputeGradGammaBeta()1242 Tensor part_grad_beta = at::native::empty_like(part_grad_gamma); in LayerNormBackwardKernelImplInternal() local