From a1388ced63dbb7e7aab4c0ad459a84ac2374777b Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Fri, 14 Oct 2016 12:41:24 -0800 Subject: Extending matrix_solve to complex64 and complex128. Change: 136192939 --- tensorflow/core/kernels/linalg_ops_common.h | 1 + 1 file changed, 1 insertion(+) (limited to 'tensorflow/core/kernels/linalg_ops_common.h') 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; using ConstMatrixMaps = gtl::InlinedVector; using MatrixMaps = gtl::InlinedVector; + using RealScalar = typename Eigen::NumTraits::Real; // Performs a single matrix computation given input matrices, and // stores the result in outputs. For batch operations, this will be called -- cgit v1.2.3