| 61 | : ::testing::TestWithParam<std::tuple<string, int>> {}; |
| 62 | |
| 63 | TEST_P(ParameterizedSlackTest, BasicTest) { |
| 64 | GrapplerItem item; |
| 65 | SetupGrapplerItem(&item); |
| 66 | |
| 67 | Slack optimizer; |
| 68 | tensorflow::RewriterConfig_CustomGraphOptimizer config; |
| 69 | (*config.mutable_parameter_map())["slack_period"].set_s( |
| 70 | std::get<0>(GetParam())); |
| 71 | TF_ASSERT_OK(optimizer.Init(&config)); |
| 72 | |
| 73 | GraphDef output; |
| 74 | TF_ASSERT_OK(optimizer.Optimize(nullptr, item, &output)); |
| 75 | ASSERT_TRUE(graph_utils::ContainsNodeWithOp("PrefetchDataset", output)); |
| 76 | NodeDef optimized_prefetch_node = |
| 77 | output.node(graph_utils::FindGraphNodeWithOp("PrefetchDataset", output)); |
| 78 | EXPECT_EQ(optimized_prefetch_node.attr().at("slack_period").i(), |
| 79 | std::get<1>(GetParam())); |
| 80 | } |
| 81 | |
| 82 | INSTANTIATE_TEST_SUITE_P(DifferentSlackEveryValues, ParameterizedSlackTest, |
| 83 | ::testing::Values(std::make_tuple("1", 1), |
nothing calls this directly
no test coverage detected