Home
last modified time | relevance | path

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 Dmulti_lamb.cu62 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 Dmulti_lamb.cu62 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 Dcs_internal_coupling.h472 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 Dcs_internal_coupling.c1775 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 Dtest_mapmri.py647 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 Dcs_gradient.c3319 + ( 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 Dcppad_ipopt_nlp.cpp529 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 DHierarchInterpPolyApproximation.cpp4676 *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()