[TF:XLA] No need to set return value in the while loop's condition.

PiperOrigin-RevId: 191809110

diff --git a/tensorflow/compiler/tf2xla/lib/while_loop.cc b/tensorflow/compiler/tf2xla/lib/while_loop.cc
index 86c02ac..495d9c6 100644 (file)
--- a/tensorflow/compiler/tf2xla/lib/while_loop.cc
+++ b/tensorflow/compiler/tf2xla/lib/while_loop.cc
@@ -54,7 +54,6 @@ xla::StatusOr<std::vector<xla::ComputationDataHandle>> XlaWhileLoop(
         auto result,
         condition_function(unpack_tuple(parameter, arity, cond_builder.get()),
                            cond_builder.get()));
-    TF_RETURN_IF_ERROR(cond_builder->SetReturnValue(result));
   }
   TF_ASSIGN_OR_RETURN(auto cond, cond_builder->Build());