Commit bea8097e authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

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
parent de465c5f
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment