STT-tensorflow/SVD at rei/fork-r2.3 - STT-tensorflow - Gitea in the EmuNest

experiments/STT-tensorflow

A. Unique TensorFlower c0944a38a4 Add an op for singular value decomposition (SVD) of a dense matrix or batches of dense matrices. This calls Eigen::JacobiSVD<Matrix, Eigen::HouseholderQRPreconditioner> which is known to be rather slow. This change is primarily intended to get the TensorFlow interfaces and functionality in place. We intend to swap out the "backend" with a higher performance algorithm implementation in the future.

This CL also contains a small refactoring of the LinearAlgebraOp base class:

1. I moved the initial processing of inputs and outputs into separate helper functions so Compute() is not so long.

2. The derived classes are now allowed to return fewer output matrix shapes (n) than the number of op outputs (m) in which case empty (shape[0]) tensors are returned for the last m-n outputs.

Fixed a few Python linter errors that were blocking presubmit.
Change: 128990912

2016-08-01 09:16:12 -07:00

2 lines

21 B

Plaintext

Raw Permalink Blame History

#include "Eigen/SVD"