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

[XLA] Reduce the default limit on the number of sweeps in eigh to 15.

Browse Source 
It is hard to find strong results about the right number of sweeps, but most authors seem to suggest values in the range of 10-15 sweeps lead to machine precision convergence.

PiperOrigin-RevId: 355850338
Change-Id: I25296b4c6913d99302c475a20e07f64a2503bf4e
This commit is contained in:
Peter Hawkins 2021-02-05 07:56:40 -08:00 committed by TensorFlower Gardener
parent bf622b7a04
commit 844002ed99
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/xla/client/lib/self_adjoint_eig.h
View File

@ -33,7 +33,7 @@ struct SelfAdjointEigResult {
};
SelfAdjointEigResult SelfAdjointEig(XlaOp a, bool lower = true,
int64 max_iter = 100, float epsilon = 1e-6);
int64 max_iter = 15, float epsilon = 1e-6);
} // namespace xla