Home
last modified time | relevance | path

Searched refs:gamma_grad_req (Results 1 – 6 of 6) sorted by relevance

/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/nn/
H A Dlayer_norm.cu627 int gamma_grad_req = req[1]; in LayerNormGradGPUContig() local
630 CHECK_NE(gamma_grad_req, kWriteInplace); in LayerNormGradGPUContig()
642 if (gamma_grad_req != kNullOp || beta_grad_req != kNullOp) { in LayerNormGradGPUContig()
652 DType* gamma_grad_ptr = (gamma_grad_req != kNullOp) ? gamma_grad.dptr<DType>() : nullptr; in LayerNormGradGPUContig()
659 if (gamma_grad_req == kAddTo && beta_grad_req != kAddTo) { in LayerNormGradGPUContig()
664 } else if (gamma_grad_req != kAddTo && beta_grad_req == kAddTo) { in LayerNormGradGPUContig()
669 } else if (gamma_grad_req == kAddTo && beta_grad_req == kAddTo) { in LayerNormGradGPUContig()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/nn/
H A Dlayer_norm.cu627 int gamma_grad_req = req[1]; in LayerNormGradGPUContig() local
630 CHECK_NE(gamma_grad_req, kWriteInplace); in LayerNormGradGPUContig()
642 if (gamma_grad_req != kNullOp || beta_grad_req != kNullOp) { in LayerNormGradGPUContig()
652 DType* gamma_grad_ptr = (gamma_grad_req != kNullOp) ? gamma_grad.dptr<DType>() : nullptr; in LayerNormGradGPUContig()
659 if (gamma_grad_req == kAddTo && beta_grad_req != kAddTo) { in LayerNormGradGPUContig()
664 } else if (gamma_grad_req != kAddTo && beta_grad_req == kAddTo) { in LayerNormGradGPUContig()
669 } else if (gamma_grad_req == kAddTo && beta_grad_req == kAddTo) { in LayerNormGradGPUContig()
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/tests/python/unittest/
H A Dtest_numpy_op.py1428 data_grad_req, gamma_grad_req, beta_grad_req): argument
1436 bn_gamma.attach_grad(grad_req=gamma_grad_req)
1456 (not fix_gamma and gamma_grad_req != 'null') or \
1512 adW = dW if gamma_grad_req != 'add' else adW + dW
1537 if gamma_grad_req != 'null':
1553 for gamma_grad_req in grad_reqs:
1554 if fix_gamma and gamma_grad_req != 'null':
1562 gamma_grad_req, beta_grad_req)
H A Dtest_operator.py1853 data_grad_req, gamma_grad_req, beta_grad_req): argument
1875 bn_gamma.attach_grad(grad_req=gamma_grad_req)
1894 (not fix_gamma and gamma_grad_req != 'null') or \
1950 adW = dW if gamma_grad_req != 'add' else adW + dW
1980 if gamma_grad_req != 'null':
1997 for gamma_grad_req in grad_reqs:
1998 if fix_gamma and gamma_grad_req != 'null':
2006 gamma_grad_req, beta_grad_req)
/dports/misc/mxnet/incubator-mxnet-1.9.0/tests/python/unittest/
H A Dtest_numpy_op.py1428 data_grad_req, gamma_grad_req, beta_grad_req): argument
1436 bn_gamma.attach_grad(grad_req=gamma_grad_req)
1456 (not fix_gamma and gamma_grad_req != 'null') or \
1512 adW = dW if gamma_grad_req != 'add' else adW + dW
1537 if gamma_grad_req != 'null':
1553 for gamma_grad_req in grad_reqs:
1554 if fix_gamma and gamma_grad_req != 'null':
1562 gamma_grad_req, beta_grad_req)
H A Dtest_operator.py1853 data_grad_req, gamma_grad_req, beta_grad_req): argument
1875 bn_gamma.attach_grad(grad_req=gamma_grad_req)
1894 (not fix_gamma and gamma_grad_req != 'null') or \
1950 adW = dW if gamma_grad_req != 'add' else adW + dW
1980 if gamma_grad_req != 'null':
1997 for gamma_grad_req in grad_reqs:
1998 if fix_gamma and gamma_grad_req != 'null':
2006 gamma_grad_req, beta_grad_req)