experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Lint fixes

Browse Source
This commit is contained in:
Evgenii Zheltonozhskii 2020-04-13 23:57:25 +03:00
parent e32a543df6
commit 1ca854ed52
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/ops/linalg_grad.py
View File

@ -688,8 +688,8 @@ def _EigGrad(op, grad_e, grad_v):
# non-diagonalizable one # non-diagonalizable one
grad_a = linalg_ops.matrix_solve(vt, grad_a = linalg_ops.matrix_solve(vt,
math_ops.matmul( math_ops.matmul(
array_ops.matrix_diag(grad_e), array_ops.matrix_diag(grad_e),
vt)) vt))
return math_ops.cast(grad_a, op.inputs[0].dtype) return math_ops.cast(grad_a, op.inputs[0].dtype)
@ops.RegisterGradient("SelfAdjointEigV2") @ops.RegisterGradient("SelfAdjointEigV2")