message: In While op, if input is TensorList, make sure output is TensorList as well.
diff: @@ -529,7 +529,11 @@ void XlaWhileOp::Compile(XlaOpKernelContext* ctx) {
  int resource_index = 0;
  for (int i = 0; i < ctx->num_outputs(); ++i) {
  if (ctx->input_type(i) != DT_RESOURCE) {
  + if (IsTensorListInput(ctx, i)) {
  + ctx->SetTensorListOutput(i, xla::GetTupleElement(while_result, i));
  + } else {
  ctx->SetOutput(i, xla::GetTupleElement(while_result, i));
  + }
  ++resource_index;
  } else {
  break;
  