Changes {batch_}self_adjoint_eig ops to return eigenvalues and eigenvectors as...
Changes {batch_}self_adjoint_eig ops to return eigenvalues and eigenvectors as separate outputs. This is implemented as a new set of kernels SelfAdjointEigV2 and BatchSelfAdjointEigV2. {batch_}self_adjoint_eig become Python wrappers that call the new kernels.
Adds new ops self_adjoint_eigvals and batch_self_adjoint_eigvals that compute the eigenvalues but not the eigenvectors, which is faster and uses less memory than computing both.
NOTICE: This changes the public API and TensorFlow models calling tf.self_adjoint_eig or tf.batch_self_adjoint_eig from Python will most likely break. Models saved as serialized graphs should be unaffected.
This CL also cleans up the python shape inference functions in linalg_ops.py a bit.
Change: 129390113
Loading
Please sign in to comment