Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/utils/
Dtraversal.h65 struct DfsPredicates { struct
66 DfsPredicates() = default;
67 DfsPredicates(std::function<bool(const NodeDef*)> enter, in DfsPredicates() argument
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
Dtraversal_test.cc188 DfsPredicates::Enter(enter), in TEST()
225 DfsPredicates::Advance(advance), in TEST()
Dtraversal.cc48 const DfsPredicates& predicates, in DfsTraversal()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc1362 DfsPredicates::Advance( in MaybeDeadOutputs()
1448 const auto predicates = DfsPredicates::Enter( in CheckThatSideEffectsWillExecute()
Dloop_optimizer.cc473 DfsPredicates::Advance(is_op_to_traverse), in GetStackPushNodesToConvert()
Dmemory_optimizer.cc1291 DfsPredicates::Advance([&](const NodeDef* node) { in RelaxAllocatorConstraints()
Darithmetic_optimizer.cc3445 DfsPredicates::Advance(is_continue_traversal), in DedupComputations()