Lines Matching refs:x_a
332 PetscScalar x_a = x[i - r.i_start]; in rosenbrock_for_loop() local
335 func(k, x_a, x_b); in rosenbrock_for_loop()
342 …_loop(r, x, o, [&](PetscInt k, PetscScalar x_a, PetscScalar x_b) { f_vec[k] = RosenbrockObjective(… in RosenbrockObjective_Kernel()
347 …ck_for_loop(r, x, o, [&](PetscInt k, PetscScalar x_a, PetscScalar x_b) { RosenbrockGradient(r.alph… in RosenbrockGradient_Kernel()
352 …p(r, x, o, [&](PetscInt k, PetscScalar x_a, PetscScalar x_b) { f_vec[k] = RosenbrockObjectiveGradi… in RosenbrockObjectiveGradient_Kernel()
357 …ck_for_loop(r, x, o, [&](PetscInt k, PetscScalar x_a, PetscScalar x_b) { RosenbrockHessian(r.alpha… in RosenbrockHessian_Kernel()
408 PetscScalar x_a = x[i - r.i_start]; in RosenbrockObjective_Host() local
411 _f += RosenbrockObjective(r.alpha, x_a, x_b); in RosenbrockObjective_Host()
423 PetscScalar x_a = x[i - r.i_start]; in RosenbrockGradient_Host() local
426 RosenbrockGradient(r.alpha, x_a, x_b, &g[2 * k]); in RosenbrockGradient_Host()
439 PetscScalar x_a = x[i - r.i_start]; in RosenbrockObjectiveGradient_Host() local
442 _f += RosenbrockObjectiveGradient(r.alpha, x_a, x_b, &g[2 * k]); in RosenbrockObjectiveGradient_Host()
454 PetscScalar x_a = x[i - r.i_start]; in RosenbrockHessian_Host() local
457 RosenbrockHessian(r.alpha, x_a, x_b, &h[4 * k]); in RosenbrockHessian_Host()