| H A D | rosenbrock4.h | 333 PetscScalar x_b = ((i + 1) < r.i_end) ? x[i + 1 - r.i_start] : o[0]; in rosenbrock_for_loop() local 335 func(k, x_a, x_b); in rosenbrock_for_loop() 342 … o, [&](PetscInt k, PetscScalar x_a, PetscScalar x_b) { f_vec[k] = RosenbrockObjective(r.alpha, x_… in RosenbrockObjective_Kernel() 347 …(r, x, o, [&](PetscInt k, PetscScalar x_a, PetscScalar x_b) { RosenbrockGradient(r.alpha, x_a, x_b… in RosenbrockGradient_Kernel() 352 …[&](PetscInt k, PetscScalar x_a, PetscScalar x_b) { f_vec[k] = RosenbrockObjectiveGradient(r.alpha… in RosenbrockObjectiveGradient_Kernel() 357 …(r, x, o, [&](PetscInt k, PetscScalar x_a, PetscScalar x_b) { RosenbrockHessian(r.alpha, x_a, x_b,… in RosenbrockHessian_Kernel() 409 PetscScalar x_b = ((i + 1) < r.i_end) ? x[i + 1 - r.i_start] : o[0]; in RosenbrockObjective_Host() local 411 _f += RosenbrockObjective(r.alpha, x_a, x_b); in RosenbrockObjective_Host() 424 PetscScalar x_b = ((i + 1) < r.i_end) ? x[i + 1 - r.i_start] : o[0]; in RosenbrockGradient_Host() local 426 RosenbrockGradient(r.alpha, x_a, x_b, &g[2 * k]); in RosenbrockGradient_Host() [all …]
|