MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / MkCallbacks

Function MkCallbacks

tensorflow/core/grappler/utils/traversal_test.cc:28–36  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

26using ::tensorflow::test::function::NDef;
27
28DfsCallbacks 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
38TEST(TraversalTest, OutputsDfsNoLoop) {
39 const string op = "OpIsNotImportantInThisTest";

Callers 1

TESTFunction · 0.85

Calls 3

nameMethod · 0.65
StrCatFunction · 0.50
push_backMethod · 0.45

Tested by

no test coverage detected