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 D | layer_norm.cu | 627 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 D | layer_norm.cu | 627 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 D | test_numpy_op.py | 1428 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 D | test_operator.py | 1853 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 D | test_numpy_op.py | 1428 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 D | test_operator.py | 1853 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)
|