Searched refs:TraversalDirection (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | traversal_test.cc | 58 DfsTraversal(graph_view, start_nodes, TraversalDirection::kFollowOutputs, in TEST()
89 DfsTraversal(graph_view, start_nodes, TraversalDirection::kFollowInputs, in TEST()
119 DfsTraversal(graph_view, start_nodes, TraversalDirection::kFollowInputs, in TEST()
150 DfsTraversal(graph_view, start_nodes, TraversalDirection::kFollowOutputs, in TEST()
187 DfsTraversal(graph_view, start_nodes, TraversalDirection::kFollowOutputs, in TEST()
224 DfsTraversal(graph_view, start_nodes, TraversalDirection::kFollowOutputs, in TEST()
|
| D | traversal.h | 26 enum class TraversalDirection { kFollowInputs, kFollowOutputs }; enum
90 TraversalDirection direction, const DfsPredicates& predicates,
98 TraversalDirection direction, const DfsCallbacks& callbacks);
|
| D | traversal.cc | 47 const TraversalDirection direction, in DfsTraversal()
107 if (direction == TraversalDirection::kFollowInputs) { in DfsTraversal()
121 TraversalDirection direction, const DfsCallbacks& callbacks) { in DfsTraversal()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | memory_optimizer.cc | 594 DfsTraversal(graph_topology, {node}, TraversalDirection::kFollowInputs, in SchedulingPass()
1290 TraversalDirection::kFollowOutputs, in RelaxAllocatorConstraints()
|
| D | function_optimizer.cc | 1360 TraversalDirection::kFollowOutputs, in MaybeDeadOutputs()
1452 TraversalDirection::kFollowOutputs, predicates, callbacks); in CheckThatSideEffectsWillExecute()
|
| D | loop_optimizer.cc | 472 TraversalDirection::kFollowOutputs, in GetStackPushNodesToConvert()
|
| D | arithmetic_optimizer.cc | 3444 DfsTraversal(graph_view, {&root}, TraversalDirection::kFollowInputs, in DedupComputations()
|