Function TEST_F

tensorflow/core/grappler/optimizers/model_pruner_test.cc:40–51

class ModelPrunerTest : public GrapplerTest {};

TEST_F(ModelPrunerTest, NoPruning) {
  // This trivial graph is so basic there's nothing to prune.
  TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"});
  GrapplerItem item;
  ASSERT_TRUE(fake_input.NextItem(&item));

  ModelPruner pruner;
  GraphDef output;
  TF_ASSERT_OK(pruner.Optimize(nullptr, item, &output));

  CompareGraphs(item.graph, output);
}

TEST_F(ModelPrunerTest, StopGradientPruning) {
  // Build a simple graph with a few trivially prunable ops.