Searched refs:GetFanouts (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | frame.cc | 53 graph_view.GetFanouts(*ready_node, /*include_controlled_nodes=*/true); in InferFromGraphView()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | model_pruner.cc | 48 graph_view.GetFanouts(node, /*include_controlled_nodes=*/true)) { in IsTrivialIdentity()
117 graph_view.GetFanouts(node, /*include_controlled_nodes=*/false)) { in CanRemoveNode()
|
| D | pin_to_host_optimizer.cc | 361 for (const GraphView::InputPort& fanout : graph.GetFanouts(*node, false)) { in Optimize()
|
| D | loop_optimizer.cc | 830 view.GetFanouts(*dead.node, true)) { in RemoveDeadBranches()
841 view.GetFanouts(*dead.node, true)) { in RemoveDeadBranches()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | graph_view_test.cc | 165 for (const auto& fo : graph.GetFanouts(*add_node, false)) { in TEST_F()
445 graph_view.GetFanouts(*node, /*include_controlled_nodes=*/false); in BM_GraphViewGetFanouts()
|
| D | graph_view.h | 206 absl::flat_hash_set<InputPort> GetFanouts( in GetFanouts() function
|
| D | mutable_graph_view_test.cc | 59 graph.GetFanouts(*node, /*include_controlled_nodes=*/true); in CompareNodeFanouts()
121 mutable_graph.GetFanouts(node, /*include_controlled_nodes=*/true); in CheckGraph()
123 immutable_graph.GetFanouts(node, /*include_controlled_nodes=*/true); in CheckGraph()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 1921 for (const auto& fanout : shape_refiner->graph().GetFanouts( in PropagateShapes()
2065 graph_view.GetFanouts(node, false)) { in InferStatically()
2069 graph_view.GetFanouts(enter, false)) { in InferStatically()
|