Home
last modified time | relevance | path

Searched refs:local_gamma_grad (Results 1 – 2 of 2) sorted by relevance

/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/nn/
H A Dlayer_norm.cu404 AType local_gamma_grad = 0; in LayerNormFusedBackwardKernel_PartGammaBeta() local
416 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()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/nn/
H A Dlayer_norm.cu404 AType local_gamma_grad = 0; in LayerNormFusedBackwardKernel_PartGammaBeta() local
416 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()