Lint fixes
This commit is contained in:
parent
e32a543df6
commit
1ca854ed52
@ -688,8 +688,8 @@ def _EigGrad(op, grad_e, grad_v):
|
||||
# non-diagonalizable one
|
||||
grad_a = linalg_ops.matrix_solve(vt,
|
||||
math_ops.matmul(
|
||||
array_ops.matrix_diag(grad_e),
|
||||
vt))
|
||||
array_ops.matrix_diag(grad_e),
|
||||
vt))
|
||||
return math_ops.cast(grad_a, op.inputs[0].dtype)
|
||||
|
||||
@ops.RegisterGradient("SelfAdjointEigV2")
|
||||
|
Loading…
Reference in New Issue
Block a user