Searched defs:part_grad_gamma (Results 1 – 1 of 1) sorted by relevance
910 T_ACC* part_grad_gamma, in cuComputePartGradGammaBeta()971 const T_ACC* part_grad_gamma, in cuComputeGradGammaBeta()1241 Tensor part_grad_gamma = at::empty({part_size,N}, gamma.options().dtype(part_grad_dtype)); in LayerNormBackwardKernelImplInternal() local