| 26 | using ::tensorflow::test::function::NDef; |
| 27 | |
| 28 | DfsCallbacks MkCallbacks(std::vector<string>* pre_order, |
| 29 | std::vector<string>* post_order, |
| 30 | std::vector<string>* back_edges) { |
| 31 | return {[pre_order](const NodeDef* n) { pre_order->push_back(n->name()); }, |
| 32 | [post_order](const NodeDef* n) { post_order->push_back(n->name()); }, |
| 33 | [back_edges](const NodeDef* src, const NodeDef* dst) { |
| 34 | back_edges->push_back(absl::StrCat(src->name(), "->", dst->name())); |
| 35 | }}; |
| 36 | } |
| 37 | |
| 38 | TEST(TraversalTest, OutputsDfsNoLoop) { |
| 39 | const string op = "OpIsNotImportantInThisTest"; |