Searched refs:DfsCallbacks (Results 1 – 7 of 7) sorted by relevance
35 struct DfsCallbacks { struct36 DfsCallbacks() = default;37 DfsCallbacks(std::function<void(const NodeDef*)> pre, in DfsCallbacks() function44 static DfsCallbacks PreOrder(std::function<void(const NodeDef*)> pre) { in PreOrder() argument45 return DfsCallbacks(std::move(pre), nullptr, nullptr); in PreOrder()48 static DfsCallbacks PostOrder(std::function<void(const NodeDef*)> post) { in PostOrder() argument49 return DfsCallbacks(nullptr, std::move(post), nullptr); in PostOrder()91 const DfsCallbacks& callbacks);98 TraversalDirection direction, const DfsCallbacks& callbacks);
49 const DfsCallbacks& callbacks) { in DfsTraversal()121 TraversalDirection direction, const DfsCallbacks& callbacks) { in DfsTraversal()
28 DfsCallbacks MkCallbacks(std::vector<string>* pre_order, in MkCallbacks()
595 DfsCallbacks::PostOrder([&topo_order](const NodeDef* n) { in SchedulingPass()1294 DfsCallbacks::PreOrder([&](const NodeDef* node) { in RelaxAllocatorConstraints()
1365 DfsCallbacks::PreOrder([&](const NodeDef* node) { in MaybeDeadOutputs()1440 const auto callbacks = DfsCallbacks::PreOrder([&](const NodeDef* node) { in CheckThatSideEffectsWillExecute()
474 DfsCallbacks::PreOrder([&](const NodeDef* node) { in GetStackPushNodesToConvert()
3446 DfsCallbacks::PreOrder([&](const NodeDef* node) { in DedupComputations()