Searched refs:DfsCallbacks (Results 1 – 6 of 6) sorted by relevance
35 struct DfsCallbacks { struct36 DfsCallbacks() = default;37 DfsCallbacks(std::function<void(const NodeDef*)> pre, in DfsCallbacks() argument44 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()
192 DfsCallbacks::PreOrder([&](const NodeDef* node) { in DedupComputations()
622 DfsCallbacks::PostOrder([&topo_order](const NodeDef* n) { in SchedulingPass()1340 DfsCallbacks::PreOrder([&](const NodeDef* node) { in FindAssignNodesToRelax()
473 DfsCallbacks::PreOrder([&](const NodeDef* node) { in GetStackPushNodesToConvert()