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()
|