Function TEST

tensorflow/core/grappler/optimizers/data/make_sloppy_test.cc:32–58

namespace {

TEST(MakeSloppy, ParallelInterleave) {
  using test::function::NDef;
  GrapplerItem item;
  item.graph = test::function::GDef(
      {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}),
       NDef("stop", "Const", {}, {{"value", 10}, {"dtype", DT_INT32}}),
       NDef("step", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}),
       NDef("range", "RangeDataset", {"start", "stop", "step"}, {}),
       NDef("cycle_length", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}),
       NDef("block_length", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}),
       NDef("num_parallel_calls", "Const", {},
            {{"value", 1}, {"dtype", DT_INT32}}),
       graph_tests_utils::MakeParallelInterleaveV2Node(
           "interleave", "range", "cycle_length", "block_length",
           "num_parallel_calls", "XTimesTwo", /*sloppy=*/false)},
      // FunctionLib
      {
          test::function::XTimesTwo(),
      });

  MakeSloppy optimizer;
  GraphDef output;
  TF_ASSERT_OK(optimizer.Optimize(nullptr, item, &output));
  EXPECT_TRUE(graph_utils::ContainsGraphNodeWithName("interleave", output));
  int index = graph_utils::FindGraphNodeWithName("interleave", output);
  EXPECT_TRUE(output.node(index).attr().at("sloppy").b());
}

TEST(MakeSloppy, ParallelMap) {
  using test::function::NDef;