Searched refs:GetFanouts (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | model_pruner.cc | 49 graph_view.GetFanouts(node, /*include_controlled_nodes=*/true)) { in IsTrivialIdentity()
135 graph_view.GetFanouts(node, /*include_controlled_nodes=*/false)) { in CanRemoveNode()
|
| D | pin_to_host_optimizer.cc | 357 for (const GraphView::InputPort& fanout : graph.GetFanouts(*node, false)) { in Optimize()
|
| D | loop_optimizer.cc | 858 view.GetFanouts(*dead.node, true)) { in RemoveDeadBranches()
869 view.GetFanouts(*dead.node, true)) { in RemoveDeadBranches()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | graph_view_test.cc | 166 for (const auto& fo : graph.GetFanouts(*add_node, false)) { in TEST_F()
448 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/optimizers/data/ |
| D | auto_shard.cc | 501 auto fanouts = graph->GetFanouts(*flat_map_node, false); in RecursivelyHandleOp()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 2354 for (const auto& fanout : shape_refiner->graph().GetFanouts( in PropagateShapes()
2500 graph_view.GetFanouts(node, false)) { in InferStatically()
2504 graph_view.GetFanouts(enter, false)) { in InferStatically()
|