| 158 | } |
| 159 | |
| 160 | static tf2xla::Config FetchesConfig(std::vector<string> fetches) { |
| 161 | tf2xla::Config config; |
| 162 | for (const auto& fetch_node_name : fetches) { |
| 163 | auto* fetch = config.add_fetch(); |
| 164 | fetch->set_name(absl::StrCat("fetch_", fetch_node_name)); |
| 165 | fetch->mutable_id()->set_node_name(fetch_node_name); |
| 166 | } |
| 167 | return config; |
| 168 | } |
| 169 | |
| 170 | TEST(PruneGraphDefInto, Basic) { |
| 171 | GraphDef def; |