diff options
Diffstat (limited to 'tensorflow/compiler/tf2xla/kernels/while_op.cc')
-rw-r--r-- | tensorflow/compiler/tf2xla/kernels/while_op.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/compiler/tf2xla/kernels/while_op.cc b/tensorflow/compiler/tf2xla/kernels/while_op.cc index 9413a30a6c..009fdd81b2 100644 --- a/tensorflow/compiler/tf2xla/kernels/while_op.cc +++ b/tensorflow/compiler/tf2xla/kernels/while_op.cc @@ -299,6 +299,7 @@ void XlaWhileOp::Compile(XlaOpKernelContext* ctx) { VLOG(1) << "Done building while loop"; } +REGISTER_XLA_OP(Name("While").AllowResourceTypes(), XlaWhileOp); REGISTER_XLA_OP(Name("XlaWhile").AllowResourceTypes(), XlaWhileOp); } // namespace tensorflow |