Lines Matching refs:o

325 … rosenbrock_for_loop(Rosenbrock r, const PetscScalar x[], const PetscScalar o[], T &&func) noexcept  in rosenbrock_for_loop()  argument
333 PetscScalar x_b = ((i + 1) < r.i_end) ? x[i + 1 - r.i_start] : o[0]; in rosenbrock_for_loop()
340 …rockObjective_Kernel(Rosenbrock r, const PetscScalar x[], const PetscScalar o[], PetscScalar f_vec… in RosenbrockObjective_Kernel() argument
342 …rosenbrock_for_loop(r, x, o, [&](PetscInt k, PetscScalar x_a, PetscScalar x_b) { f_vec[k] = Rosenb… in RosenbrockObjective_Kernel()
345 …enbrockGradient_Kernel(Rosenbrock r, const PetscScalar x[], const PetscScalar o[], PetscScalar g[]) in RosenbrockGradient_Kernel() argument
347 …rosenbrock_for_loop(r, x, o, [&](PetscInt k, PetscScalar x_a, PetscScalar x_b) { RosenbrockGradien… in RosenbrockGradient_Kernel()
350 …ctiveGradient_Kernel(Rosenbrock r, const PetscScalar x[], const PetscScalar o[], PetscScalar f_vec… in RosenbrockObjectiveGradient_Kernel() argument
352 …rosenbrock_for_loop(r, x, o, [&](PetscInt k, PetscScalar x_a, PetscScalar x_b) { f_vec[k] = Rosenb… in RosenbrockObjectiveGradient_Kernel()
355 …senbrockHessian_Kernel(Rosenbrock r, const PetscScalar x[], const PetscScalar o[], PetscScalar h[]) in RosenbrockHessian_Kernel() argument
357 …rosenbrock_for_loop(r, x, o, [&](PetscInt k, PetscScalar x_a, PetscScalar x_b) { RosenbrockHessian… in RosenbrockHessian_Kernel()
360 …cupmStream_t stream, Rosenbrock r, const PetscScalar x[], const PetscScalar o[], PetscScalar f_vec… in RosenbrockObjective_Device() argument
365 …scCUPMLaunch(RosenbrockObjective_Kernel<<<(n_comp + 255) / 256, 256, 0, stream>>>(r, x, o, f_vec)); in RosenbrockObjective_Device()
370 …e(cupmStream_t stream, Rosenbrock r, const PetscScalar x[], const PetscScalar o[], PetscScalar g[]) in RosenbrockGradient_Device() argument
375 …) PetscCUPMLaunch(RosenbrockGradient_Kernel<<<(n_comp + 255) / 256, 256, 0, stream>>>(r, x, o, g)); in RosenbrockGradient_Device()
380 …cupmStream_t stream, Rosenbrock r, const PetscScalar x[], const PetscScalar o[], PetscScalar f_vec… in RosenbrockObjectiveGradient_Device() argument
385 …h(RosenbrockObjectiveGradient_Kernel<<<(n_comp + 255) / 256, 256, 0, stream>>>(r, x, o, f_vec, g)); in RosenbrockObjectiveGradient_Device()
390 …e(cupmStream_t stream, Rosenbrock r, const PetscScalar x[], const PetscScalar o[], PetscScalar h[]) in RosenbrockHessian_Device() argument
395 …p) PetscCUPMLaunch(RosenbrockHessian_Kernel<<<(n_comp + 255) / 256, 256, 0, stream>>>(r, x, o, h)); in RosenbrockHessian_Device()
401 … RosenbrockObjective_Host(Rosenbrock r, const PetscScalar x[], const PetscScalar o[], PetscReal *f) in RosenbrockObjective_Host() argument
409 PetscScalar x_b = ((i + 1) < r.i_end) ? x[i + 1 - r.i_start] : o[0]; in RosenbrockObjective_Host()
418 …osenbrockGradient_Host(Rosenbrock r, const PetscScalar x[], const PetscScalar o[], PetscScalar g[]) in RosenbrockGradient_Host() argument
424 PetscScalar x_b = ((i + 1) < r.i_end) ? x[i + 1 - r.i_start] : o[0]; in RosenbrockGradient_Host()
432 …jectiveGradient_Host(Rosenbrock r, const PetscScalar x[], const PetscScalar o[], PetscReal *f, Pet… in RosenbrockObjectiveGradient_Host() argument
440 PetscScalar x_b = ((i + 1) < r.i_end) ? x[i + 1 - r.i_start] : o[0]; in RosenbrockObjectiveGradient_Host()
449 …RosenbrockHessian_Host(Rosenbrock r, const PetscScalar x[], const PetscScalar o[], PetscScalar h[]) in RosenbrockHessian_Host() argument
455 PetscScalar x_b = ((i + 1) < r.i_end) ? x[i + 1 - r.i_start] : o[0]; in RosenbrockHessian_Host()
470 const PetscScalar *o = NULL; in FormObjective() local
477 PetscCall(VecGetArrayReadAndMemType(user->off_process_values, &o, NULL)); in FormObjective()
480 PetscCall(RosenbrockObjective_Host(user->problem, x, o, &f_local)); in FormObjective()
492 PetscCall(RosenbrockObjective_Device(*stream, user->problem, x, o, _fvec)); in FormObjective()
499 PetscCall(VecRestoreArrayReadAndMemType(user->off_process_values, &o)); in FormObjective()
509 const PetscScalar *o = NULL; in FormGradient() local
516 PetscCall(VecGetArrayReadAndMemType(user->off_process_values, &o, NULL)); in FormGradient()
521 PetscCall(RosenbrockGradient_Host(user->problem, x, o, g)); in FormGradient()
529 PetscCall(RosenbrockGradient_Device(*stream, user->problem, x, o, g)); in FormGradient()
534 PetscCall(VecRestoreArrayReadAndMemType(user->off_process_values, &o)); in FormGradient()
565 const PetscScalar *o = NULL; in FormObjectiveGradient() local
572 PetscCall(VecGetArrayReadAndMemType(user->off_process_values, &o, NULL)); in FormObjectiveGradient()
577 PetscCall(RosenbrockObjectiveGradient_Host(user->problem, x, o, &f_local, g)); in FormObjectiveGradient()
589 PetscCall(RosenbrockObjectiveGradient_Device(*stream, user->problem, x, o, _fvec, g)); in FormObjectiveGradient()
598 PetscCall(VecRestoreArrayReadAndMemType(user->off_process_values, &o)); in FormObjectiveGradient()
626 const PetscScalar *o = NULL; in FormHessian() local
632 PetscCall(VecGetArrayReadAndMemType(user->off_process_values, &o, NULL)); in FormHessian()
637 PetscCall(RosenbrockHessian_Host(user->problem, x, o, h)); in FormHessian()
645 PetscCall(RosenbrockHessian_Device(*stream, user->problem, x, o, h)); in FormHessian()
653 PetscCall(VecRestoreArrayReadAndMemType(user->off_process_values, &o)); in FormHessian()