Home
last modified time | relevance | path

Searched defs:buf_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.cu402 AType* buf_gamma_grad = d_buf; in LayerNormFusedBackwardKernel_PartGammaBeta() local
453 AType* buf_gamma_grad = reinterpret_cast<AType*>(buf); in LayerNormFusedBackwardKernel_GammaBeta() local
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/nn/
H A Dlayer_norm.cu402 AType* buf_gamma_grad = d_buf; in LayerNormFusedBackwardKernel_PartGammaBeta() local
453 AType* buf_gamma_grad = reinterpret_cast<AType*>(buf); in LayerNormFusedBackwardKernel_GammaBeta() local