commit | 6777d627460a0fcaa0d9f8f91d01373dbfd3dc21 | [log] [tgz] |
---|---|---|
author | David Majnemer <majnemer@google.com> | Thu Jul 08 09:41:52 2021 -0700 |
committer | TensorFlower Gardener <gardener@tensorflow.org> | Thu Jul 08 09:47:41 2021 -0700 |
tree | 259173e278de10cc52a829732d2b19dd796b961a | |
parent | c30dcc957ccb245f35d4956c5aa9cec73b12c740 [diff] |
Remove stale limitations PiperOrigin-RevId: 383652551 Change-Id: I68fc5be954d4f02943d6f2545a61893120b569ec
diff --git a/tensorflow/compiler/tf2xla/kernels/cholesky_op.cc b/tensorflow/compiler/tf2xla/kernels/cholesky_op.cc index e6b30a3..0c3103b 100644 --- a/tensorflow/compiler/tf2xla/kernels/cholesky_op.cc +++ b/tensorflow/compiler/tf2xla/kernels/cholesky_op.cc
@@ -31,7 +31,8 @@ } }; -REGISTER_XLA_OP(Name("Cholesky").TypeConstraint("T", kFloatTypes), CholeskyOp); +REGISTER_XLA_OP(Name("Cholesky").TypeConstraint("T", kFloatAndComplexTypes), + CholeskyOp); } // namespace } // namespace tensorflow