Lines Matching refs:state
69 output: A namedtuple representing the final state with fields:
76 # ephemeral class holding CGLS state.
80 def stopping_criterion(i, state):
81 return math_ops.logical_and(i < max_iter, state.gamma > tol)
84 def cgls_step(i, state):
85 q = operator.apply(state.p)
86 alpha = state.gamma / util.l2norm_squared(q)
87 x = state.x + alpha * state.p
88 r = state.r - alpha * q
91 beta = gamma / state.gamma
92 p = s + beta * state.p
105 state = cgls_state(i=i, x=x, r=rhs, p=s0, gamma=gamma0)
106 _, state = control_flow_ops.while_loop(stopping_criterion, cgls_step,
107 [i, state])
109 state.i,
110 x=array_ops.squeeze(state.x),
111 r=array_ops.squeeze(state.r),
112 p=array_ops.squeeze(state.p),
113 gamma=state.gamma)
