diff options
Diffstat (limited to 'tensorflow/core/ops/linalg_ops.cc')
-rw-r--r-- | tensorflow/core/ops/linalg_ops.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/ops/linalg_ops.cc b/tensorflow/core/ops/linalg_ops.cc index a2762cf206..872824b885 100644 --- a/tensorflow/core/ops/linalg_ops.cc +++ b/tensorflow/core/ops/linalg_ops.cc @@ -318,7 +318,7 @@ REGISTER_OP("SelfAdjointEigV2") .Output("e: T") .Output("v: T") .Attr("compute_v: bool = True") - .Attr("T: {double, float}") + .Attr("T: {double, float, complex64, complex128}") .SetShapeFn(SelfAdjointEigV2ShapeFn) .Doc(R"doc( Computes the eigen decomposition of one or more square self-adjoint matrices. |