Searched refs:local_gamma_grad (Results 1 – 2 of 2) sorted by relevance
404 AType local_gamma_grad = 0; in LayerNormFusedBackwardKernel_PartGammaBeta() local416 local_gamma_grad += (local_in_data - local_mean) / local_std * local_out_grad; in LayerNormFusedBackwardKernel_PartGammaBeta()421 buf_gamma_grad[threadIdx.y * row_stride + threadIdx.x] = local_gamma_grad; in LayerNormFusedBackwardKernel_PartGammaBeta()