def compute(self, value):
      return self.initialized_variable + value

  to_save = Module()
  saved_model.save(
      to_save, export_dir=os.path.join(base_dir, "UninitializedVariable"))


def _gen_simple_while_loop(base_dir):
  """Generates a saved model with a while loop."""

  class Module(module.Module):
    """A module with a while loop."""

    @def_function.function(