Searched refs:kFollowInputs (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | pattern_utils_test.cc | 174 SubGraphMatcher<MatchingDirection::kFollowInputs> graph_matcher(&graph_view); in TEST_F()
256 SubGraphMatcher<MatchingDirection::kFollowInputs> graph_matcher(&graph_view); in TEST_F()
342 SubGraphMatcher<MatchingDirection::kFollowInputs> graph_matcher(&graph_view); in TEST_F()
363 SubGraphMatcher<MatchingDirection::kFollowInputs> graph_matcher(&graph_view); in TEST_F()
399 SubGraphMatcher<MatchingDirection::kFollowInputs> graph_matcher(&graph_view); in TEST_F()
420 SubGraphMatcher<MatchingDirection::kFollowInputs> graph_matcher(&graph_view); in TEST_F()
|
| D | pattern_utils.h | 121 enum class MatchingDirection { kFollowInputs, kFollowOutputs }; enumerator
178 template <MatchingDirection DIRECTION = MatchingDirection::kFollowInputs>
|
| D | pattern_utils.cc | 27 bool SubGraphMatcher<MatchingDirection::kFollowInputs>::DoesOpTypePatternMatch( in DoesOpTypePatternMatch()
103 bool SubGraphMatcher<MatchingDirection::kFollowInputs>::GetMatchedNodes( in GetMatchedNodes()
|
| D | traversal_test.cc | 89 DfsTraversal(graph_view, start_nodes, TraversalDirection::kFollowInputs, in TEST()
119 DfsTraversal(graph_view, start_nodes, TraversalDirection::kFollowInputs, in TEST()
|
| D | traversal.h | 26 enum class TraversalDirection { kFollowInputs, kFollowOutputs }; enumerator
|
| D | traversal.cc | 107 if (direction == TraversalDirection::kFollowInputs) { in DfsTraversal()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | common_subgraph_elimination.cc | 190 DfsTraversal(graph_view, {&root}, TraversalDirection::kFollowInputs, in DedupComputations()
|
| D | auto_mixed_precision.cc | 672 kFollowInputs, enumerator
809 if (direction == TypeTraversalDirection::kFollowInputs || in DfsTypeTraversal()
1512 graph_type_view_, {&root}, TypeTraversalDirection::kFollowInputs, in FindTensorListImplicitFloat32Edges()
1572 TypeTraversalDirection::kFollowInputs, in PropagateDenyFwdThroughClearAndInfer()
1646 graph_type_view_, {&root}, TypeTraversalDirection::kFollowInputs, in AddClearAndInferToAllowIfBetweenAllow()
|
| D | memory_optimizer.cc | 621 DfsTraversal(graph_topology, {node}, TraversalDirection::kFollowInputs, in SchedulingPass()
|