Lines Matching refs:Ax
49 PetscCall(VecNorm(am->Ax, NORM_2, &Axnorm)); in TaoADMMToleranceUpdate()
77 PetscCall(VecWAXPY(tempJR, -1., am->Axold, am->Ax)); in AdaptiveADMMPenaltyUpdate()
152 static PetscErrorCode ADMMUpdateConstraintResidualVector(Tao tao, Vec x, Vec z, Vec Ax, Vec Bz, Vec… in ADMMUpdateConstraintResidualVector() argument
161 PetscCall(MatMult(mis->jacobian_equality, x, Ax)); in ADMMUpdateConstraintResidualVector()
165 PetscCall(VecWAXPY(residual, 1., Bz, Ax)); in ADMMUpdateConstraintResidualVector()
182 …PetscCall(ADMMUpdateConstraintResidualVector(parent, x, am->subsolverZ->solution, am->Ax, am->Bz, … in SubObjGradUpdate()
211 …PetscCall(ADMMUpdateConstraintResidualVector(parent, am->subsolverX->solution, z, am->Ax, am->Bz, … in RegObjGradUpdate()
414 PetscCall(MatMult(am->subsolverX->jacobian_equality, am->subsolverX->solution, am->Ax)); in TaoSolve_ADMM()
425 PetscCall(VecWAXPY(am->workJacobianRight, 1 / am->mu, am->y, am->Ax)); in TaoSolve_ADMM()
431 …esidualVector(tao, am->subsolverX->solution, am->subsolverZ->solution, am->Ax, am->Bz, am->residua… in TaoSolve_ADMM()
447 PetscCall(VecWAXPY(am->residual, 1., am->Ax, am->Bzold)); in TaoSolve_ADMM()
450 PetscCall(VecCopy(am->Ax, am->Axold)); in TaoSolve_ADMM()
455 PetscCall(VecWAXPY(am->residual, 1., am->Ax, am->Bzold)); in TaoSolve_ADMM()
459 PetscCall(VecCopy(am->Ax, am->Axold)); in TaoSolve_ADMM()
486 …esidualVector(tao, am->subsolverX->solution, am->subsolverZ->solution, am->Ax, am->Bz, am->residua… in TaoSolve_ADMM()
560 PetscCall(MatCreateVecs(am->JA, NULL, &am->Ax)); in TaoSetUp_ADMM()
569 if (!am->Axold) PetscCall(VecDuplicate(am->Ax, &am->Axold)); in TaoSetUp_ADMM()
570 if (!am->workJacobianRight) PetscCall(VecDuplicate(am->Ax, &am->workJacobianRight)); in TaoSetUp_ADMM()
571 if (!am->workJacobianRight2) PetscCall(VecDuplicate(am->Ax, &am->workJacobianRight2)); in TaoSetUp_ADMM()
572 if (!am->Bz) PetscCall(VecDuplicate(am->Ax, &am->Bz)); in TaoSetUp_ADMM()
573 if (!am->Bzold) PetscCall(VecDuplicate(am->Ax, &am->Bzold)); in TaoSetUp_ADMM()
574 if (!am->Bz0) PetscCall(VecDuplicate(am->Ax, &am->Bz0)); in TaoSetUp_ADMM()
576 PetscCall(VecDuplicate(am->Ax, &am->y)); in TaoSetUp_ADMM()
580 PetscCall(VecDuplicate(am->Ax, &am->yold)); in TaoSetUp_ADMM()
584 PetscCall(VecDuplicate(am->Ax, &am->y0)); in TaoSetUp_ADMM()
588 PetscCall(VecDuplicate(am->Ax, &am->yhat)); in TaoSetUp_ADMM()
592 PetscCall(VecDuplicate(am->Ax, &am->yhatold)); in TaoSetUp_ADMM()
596 PetscCall(VecDuplicate(am->Ax, &am->residual)); in TaoSetUp_ADMM()
623 PetscCall(VecDestroy(&am->Ax)); in TaoDestroy_ADMM()