Home
last modified time | relevance | path

Searched refs:GetFanoutEdges (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dgraph_view.h275 absl::flat_hash_set<Edge> GetFanoutEdges( in GetFanoutEdges() function
Dgraph_view_test.cc478 graph_view.GetFanoutEdges(*node, /*include_controlled_edges=*/false); in BM_GraphViewGetFanoutEdges()
Dmutable_graph_view_test.cc137 mutable_graph.GetFanoutEdges(node, /*include_controlled_edges=*/true); in CheckGraph()
139 immutable_graph.GetFanoutEdges(node, /*include_controlled_edges=*/true); in CheckGraph()
Dmutable_graph_view.cc795 GetFanoutEdges(*from_node, /*include_controlled_edges=*/false); in UpdateFanoutsInternal()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc457 for (GraphView::Edge edge : ctx.graph_view.GetFanoutEdges(*node, false)) { in FindFusedBatchNorm()
Dfunction_optimizer.cc497 ctx.graph_view().GetFanoutEdges(node, /*include_controlled_edges=*/false); in GetActiveOutputs()