Searched refs:TraversalDirection (Results 1 – 6 of 6) 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 | common_subgraph_elimination.cc | 190 DfsTraversal(graph_view, {&root}, TraversalDirection::kFollowInputs, in DedupComputations()
|
| D | memory_optimizer.cc | 621 DfsTraversal(graph_topology, {node}, TraversalDirection::kFollowInputs, in SchedulingPass()
1336 TraversalDirection::kFollowOutputs, in FindAssignNodesToRelax()
|
| D | loop_optimizer.cc | 471 TraversalDirection::kFollowOutputs, in GetStackPushNodesToConvert()
|