Home
last modified time | relevance | path

Searched refs:DfsCallbacks (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/utils/
Dtraversal.h35 struct DfsCallbacks { struct
36 DfsCallbacks() = default;
37 DfsCallbacks(std::function<void(const NodeDef*)> pre, in DfsCallbacks() function
44 static DfsCallbacks PreOrder(std::function<void(const NodeDef*)> pre) { in PreOrder() argument
45 return DfsCallbacks(std::move(pre), nullptr, nullptr); in PreOrder()
48 static DfsCallbacks PostOrder(std::function<void(const NodeDef*)> post) { in PostOrder() argument
49 return DfsCallbacks(nullptr, std::move(post), nullptr); in PostOrder()
91 const DfsCallbacks& callbacks);
98 TraversalDirection direction, const DfsCallbacks& callbacks);
Dtraversal.cc49 const DfsCallbacks& callbacks) { in DfsTraversal()
121 TraversalDirection direction, const DfsCallbacks& callbacks) { in DfsTraversal()
Dtraversal_test.cc28 DfsCallbacks MkCallbacks(std::vector<string>* pre_order, in MkCallbacks()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmemory_optimizer.cc595 DfsCallbacks::PostOrder([&topo_order](const NodeDef* n) { in SchedulingPass()
1294 DfsCallbacks::PreOrder([&](const NodeDef* node) { in RelaxAllocatorConstraints()
Dfunction_optimizer.cc1365 DfsCallbacks::PreOrder([&](const NodeDef* node) { in MaybeDeadOutputs()
1440 const auto callbacks = DfsCallbacks::PreOrder([&](const NodeDef* node) { in CheckThatSideEffectsWillExecute()
Dloop_optimizer.cc474 DfsCallbacks::PreOrder([&](const NodeDef* node) { in GetStackPushNodesToConvert()
Darithmetic_optimizer.cc3446 DfsCallbacks::PreOrder([&](const NodeDef* node) { in DedupComputations()