Searched refs:DfsPredicates (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | traversal.h | 65 struct DfsPredicates { struct
66 DfsPredicates() = default;
67 DfsPredicates(std::function<bool(const NodeDef*)> enter, in DfsPredicates() function
71 static DfsPredicates Enter(std::function<bool(const NodeDef*)> enter) { in Enter() argument
72 return DfsPredicates(std::move(enter), nullptr); in Enter()
75 static DfsPredicates Advance(std::function<bool(const NodeDef*)> advance) { in Advance() argument
76 return DfsPredicates(nullptr, std::move(advance)); in Advance()
90 TraversalDirection direction, const DfsPredicates& predicates, argument
|
| D | traversal_test.cc | 188 DfsPredicates::Enter(enter), in TEST()
225 DfsPredicates::Advance(advance), in TEST()
|
| D | traversal.cc | 48 const DfsPredicates& predicates, in DfsTraversal()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | common_subgraph_elimination.cc | 191 DfsPredicates::Advance(is_continue_traversal), in DedupComputations()
|
| D | loop_optimizer.cc | 472 DfsPredicates::Advance(is_op_to_traverse), in GetStackPushNodesToConvert()
|
| D | memory_optimizer.cc | 1337 DfsPredicates::Advance([&](const NodeDef* node) { in FindAssignNodesToRelax()
|