diff options
author | 2016-10-14 12:41:24 -0800 | |
---|---|---|
committer | 2016-10-14 13:47:34 -0700 | |
commit | a1388ced63dbb7e7aab4c0ad459a84ac2374777b (patch) | |
tree | 1065064c60af6fc6b6b59b127710da6e6d2c384d /tensorflow/core/kernels/linalg_ops_common.h | |
parent | 47318618b24d493877668750f84d114179120ca5 (diff) |
Extending matrix_solve to complex64 and complex128.
Change: 136192939
Diffstat (limited to 'tensorflow/core/kernels/linalg_ops_common.h')
-rw-r--r-- | tensorflow/core/kernels/linalg_ops_common.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/linalg_ops_common.h b/tensorflow/core/kernels/linalg_ops_common.h index e6c447a623..8a606ed0a9 100644 --- a/tensorflow/core/kernels/linalg_ops_common.h +++ b/tensorflow/core/kernels/linalg_ops_common.h @@ -114,6 +114,7 @@ class LinearAlgebraOp : public OpKernel { using MatrixMap = Eigen::Map<Matrix>; using ConstMatrixMaps = gtl::InlinedVector<ConstMatrixMap, 4>; using MatrixMaps = gtl::InlinedVector<MatrixMap, 4>; + using RealScalar = typename Eigen::NumTraits<Scalar>::Real; // Performs a single matrix computation given input matrices, and // stores the result in outputs. For batch operations, this will be called |