"//tensorflow/python/saved_model",
        "@absl_py//absl:app",
    ],
)

# Files generated by the binary above.
filegroup(
    name = "saved_models",
    srcs = glob([
        "SimpleWhileLoop/**",
        "UninitializedVariable/**",
    ]),
    visibility = [
        "//tensorflow/c/experimental/saved_model/internal:__pkg__",
    ],

          body=lambda acc, i: (acc + i, i - 1),
          loop_vars=(constant_op.constant(0.0), value))
      return acc

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


def main(args):
  if len(args) != 2:
    raise app.UsageError("Expected one argument (base_dir).")
  _, base_dir = args
  _gen_uninitialized_variable(base_dir)
  _gen_simple_while_loop(base_dir)

  TFE_DeleteContextOptions(opts);

  std::string model_dir = tensorflow::io::JoinPath(
      tensorflow::testing::TensorFlowSrcRoot(),
      "c/experimental/saved_model/internal/testdata/SimpleWhileLoop");

  TF_SavedModel* saved_model =
      TF_LoadSavedModel(model_dir.c_str(), ctx, status);
  ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);

  TF_ConcreteFunction* while_fn =