Lines Matching refs:grad_b
110 grad_b = _TriangularSolve(dq - math_ops.matmul(q, qdq), r)
111 return grad_a + grad_b
120 grad_b = linalg_ops.matrix_solve(a, grad, adjoint=not adjoint_a)
122 grad_a = -math_ops.matmul(c, grad_b, adjoint_b=True)
124 grad_a = -math_ops.matmul(grad_b, c, adjoint_b=True)
125 return (grad_a, grad_b)
160 grad_b = math_ops.matmul(a, z)
161 return (grad_a, grad_b, None)
179 grad_b = linalg_ops.cholesky_solve(chol, math_ops.matmul(a, grad))
183 a1 = -math_ops.matmul(grad_b, a1)
184 a2 = grad - math_ops.matmul(a, grad_b, adjoint_a=True)
187 return (grad_a, grad_b, None)
214 grad_b = linalg_ops.matrix_triangular_solve(
217 grad_a = -math_ops.matmul(c, grad_b, adjoint_b=True)
219 grad_a = -math_ops.matmul(grad_b, c, adjoint_b=True)
224 return (grad_a, grad_b)
