Searched refs:r_grad (Results 1 – 8 of 8) sorted by relevance
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/contrib/ |
H A D | multi_lamb.cu | 62 MPDType r_grad[ILP_LAMB]; in KernelStep1() local 75 r_grad[ii] = static_cast<MPDType>(kernel_params.grads[tensor_id][load_pos]); in KernelStep1() 80 r_grad[ii] = static_cast<MPDType>(0); in KernelStep1() 87 r_grad[ii] = r_grad[ii] * static_cast<MPDType>(rescale_grad); in KernelStep1() 89 r_grad[ii] = max(min(r_grad[ii], clip_gradient), -clip_gradient); in KernelStep1() 90 r_mean[ii] = static_cast<MPDType>(beta1) * r_mean[ii] + beta3 * r_grad[ii]; in KernelStep1() 91 r_var[ii] = static_cast<MPDType>(beta2) * r_var[ii] + beta4 * r_grad[ii] * r_grad[ii]; in KernelStep1()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/contrib/ |
H A D | multi_lamb.cu | 62 MPDType r_grad[ILP_LAMB]; in KernelStep1() local 75 r_grad[ii] = static_cast<MPDType>(kernel_params.grads[tensor_id][load_pos]); in KernelStep1() 80 r_grad[ii] = static_cast<MPDType>(0); in KernelStep1() 87 r_grad[ii] = r_grad[ii] * static_cast<MPDType>(rescale_grad); in KernelStep1() 89 r_grad[ii] = max(min(r_grad[ii], clip_gradient), -clip_gradient); in KernelStep1() 90 r_mean[ii] = static_cast<MPDType>(beta1) * r_mean[ii] + beta3 * r_grad[ii]; in KernelStep1() 91 r_var[ii] = static_cast<MPDType>(beta2) * r_var[ii] + beta4 * r_grad[ii] * r_grad[ii]; in KernelStep1()
|
/dports/science/code_saturne/code_saturne-7.1.0/src/base/ |
H A D | cs_internal_coupling.h | 472 cs_real_3_t *restrict r_grad, 489 cs_real_33_t *restrict r_grad, 506 cs_real_63_t *restrict r_grad,
|
H A D | cs_internal_coupling.c | 1775 cs_real_3_t r_grad[restrict], in cs_internal_coupling_reconstruct_scalar_gradient() 1796 (const cs_real_t *)r_grad, in cs_internal_coupling_reconstruct_scalar_gradient() 1809 rfac *= offset_vect[ii][0]*(r_grad_local[ii][0]+r_grad[cell_id][0]) in cs_internal_coupling_reconstruct_scalar_gradient() 1810 +offset_vect[ii][1]*(r_grad_local[ii][1]+r_grad[cell_id][1]) in cs_internal_coupling_reconstruct_scalar_gradient() 1811 +offset_vect[ii][2]*(r_grad_local[ii][2]+r_grad[cell_id][2]); in cs_internal_coupling_reconstruct_scalar_gradient() 1834 cs_real_33_t *restrict r_grad, in cs_internal_coupling_reconstruct_vector_gradient() argument 1855 (const cs_real_t *)r_grad, in cs_internal_coupling_reconstruct_vector_gradient() 1869 rfac *= offset_vect[ii][0]*(r_grad_local[ii][i][0]+r_grad[cell_id][i][0]) in cs_internal_coupling_reconstruct_vector_gradient() 1870 +offset_vect[ii][1]*(r_grad_local[ii][i][1]+r_grad[cell_id][i][1]) in cs_internal_coupling_reconstruct_vector_gradient() 1895 cs_real_63_t *restrict r_grad, in cs_internal_coupling_reconstruct_tensor_gradient() argument [all …]
|
/dports/science/py-dipy/dipy-1.4.1/dipy/reconst/tests/ |
H A D | test_mapmri.py | 647 r_grad = mapmri.create_rspace(gridsize, max_radius) 657 pdf = mapfit_no_constraint.pdf(r_grad) 666 pdf = mapfit_constraint.pdf(r_grad) 678 pdf = mapfit_no_constraint.pdf(r_grad) 688 pdf = mapfit_constraint.pdf(r_grad)
|
/dports/science/code_saturne/code_saturne-7.1.0/src/alge/ |
H A D | cs_gradient.c | 3319 + ( dofij[f_id][0] * (r_grad[c_id1][0]+r_grad[c_id2][0]) in _reconstruct_scalar_gradient() 3320 + dofij[f_id][1] * (r_grad[c_id1][1]+r_grad[c_id2][1]) in _reconstruct_scalar_gradient() 3321 + dofij[f_id][2] * (r_grad[c_id1][2]+r_grad[c_id2][2])) * 0.5; in _reconstruct_scalar_gradient() 3422 (dofij[f_id][0]*(r_grad[c_id1][0]+r_grad[c_id2][0]) in _reconstruct_scalar_gradient() 3423 +dofij[f_id][1]*(r_grad[c_id1][1]+r_grad[c_id2][1]) in _reconstruct_scalar_gradient() 3424 +dofij[f_id][2]*(r_grad[c_id1][2]+r_grad[c_id2][2])); in _reconstruct_scalar_gradient() 7390 cs_real_3_t *restrict r_grad; in _gradient_scalar() local 7416 r_grad); in _gradient_scalar() 7423 var, r_grad); in _gradient_scalar() 7435 r_grad, in _gradient_scalar() [all …]
|
/dports/math/cppad/CppAD-20210000.8/cppad_ipopt/src/ |
H A D | cppad_ipopt_nlp.cpp | 529 NumberVector r_grad(q_[k]); in eval_grad_f() local 538 r_grad = r_fun_[k].Reverse(1, w); in eval_grad_f() 541 grad_f[ J_[j] ] += r_grad[j]; in eval_grad_f()
|
/dports/science/dakota/dakota-6.13.0-release-public.src-UI/packages/pecos/src/ |
H A D | HierarchInterpPolyApproximation.cpp | 4676 *r_grad = r_grads[0][0][0], in integrate_response_moments() local 4679 mom_t2_coeffs_000[v] = deriv * r_grad[v]; in integrate_response_moments() 4701 *r_grad = r_grads_ls[pt], in integrate_response_moments() local 4706 mom_t2_coeffs_lsp[v] = deriv * r_grad[v] - prev_grad[v]; in integrate_response_moments()
|