Searched refs:f_vec (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/tao/unconstrained/tutorials/ |
| H A D | rosenbrock4.h | 340 …ckObjective_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()
350 …l(Rosenbrock r, const PetscScalar x[], const PetscScalar o[], PetscScalar f_vec[], PetscScalar g[]) 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()
360 …pmStream_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()
380 …, Rosenbrock r, const PetscScalar x[], const PetscScalar o[], PetscScalar f_vec[], PetscScalar g[]) in RosenbrockObjectiveGradient_Device() argument
385 …h(RosenbrockObjectiveGradient_Kernel<<<(n_comp + 255) / 256, 256, 0, stream>>>(r, x, o, f_vec, g)); in RosenbrockObjectiveGradient_Device()
|