Home
last modified time | relevance | path

Searched refs:GetFanout (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dgraph_topology_view_test.cc89 EXPECT_EQ(graph_view.GetFanout(0), Fanout({2, 3})); in TEST_F()
90 EXPECT_EQ(graph_view.GetFanout(1), Fanout({2})); in TEST_F()
91 EXPECT_EQ(graph_view.GetFanout(2), Fanout({3})); in TEST_F()
92 EXPECT_EQ(graph_view.GetFanout(3), Fanout()); in TEST_F()
112 EXPECT_EQ(graph_view.GetFanout(2), Fanout({3})); in TEST_F()
113 EXPECT_EQ(graph_view.GetFanout(3), Fanout({2})); in TEST_F()
134 EXPECT_EQ(graph_view.GetFanout(2), Fanout({3})); in TEST_F()
135 EXPECT_EQ(graph_view.GetFanout(3), Fanout({2})); in TEST_F()
147 EXPECT_EQ(graph_view.GetFanout(2), Fanout({3})); in TEST_F()
148 EXPECT_EQ(graph_view.GetFanout(3), Fanout({})); in TEST_F()
Dgraph_view.cc69 return graph_view.GetFanout(output).size() <= 1; in HasSingleFanoutNode()
74 return !graph_view.GetFanout(output).empty(); in HasFanouts()
84 return !graph_view.GetFanout(control_port).empty(); in HasControlFanout()
Dmutable_graph_view.h51 using GraphViewInternal::GetFanout;
52 const absl::flat_hash_set<InputPort>& GetFanout(
Dgraph_view_test.cc150 EXPECT_EQ(graph.GetFanout(output).size(), 2); in TEST_F()
151 for (auto fanout : graph.GetFanout(output)) { in TEST_F()
194 auto fanout = graph.GetFanout(output); in TEST_F()
202 fanout = graph.GetFanout(output); in TEST_F()
396 graph_view.GetFanout({node, 0}); in BM_GraphViewGetFanout()
Dgraph_topology_view.h90 const absl::InlinedVector<int, 2>& GetFanout(int node_idx) const;
Dgraph_topology_view.cc202 const absl::InlinedVector<int, 2>& GraphTopologyView::GetFanout( in GetFanout() function in tensorflow::grappler::GraphTopologyView
Dmutable_graph_view.cc417 MutableGraphView::GetFanout(const GraphView::OutputPort& port) const { in GetFanout() function in tensorflow::grappler::MutableGraphView
418 return GetFanout(MutableGraphView::OutputPort(const_cast<NodeDef*>(port.node), in GetFanout()
514 auto control_fanouts = GetFanout(control_port); in UpdateNode()
769 GetFanout(GraphView::OutputPort(from_node, Graph::kControlSlot)); in UpdateFanoutsInternal()
973 for (const auto& fanout : GetFanout(fanin)) { in GetControllingFaninToAdd()
Dgraph_view.h139 const absl::flat_hash_set<InputPort>& GetFanout( in GetFanout() function
Dmutable_graph_view_test.cc52 EXPECT_TRUE(graph.GetFanout(output_port).contains(input_port)); in CompareNodeFanins()
1188 auto fanouts = graph.GetFanout(output_port); in CheckFanoutRemoved()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dlatency_all_edges.cc85 auto fanout = graph.GetFanout(output_port); in OptimizeAndCollectStats()
/external/tensorflow/tensorflow/core/grappler/utils/
Dtraversal.cc112 for (const int fanout : graph_view.GetFanout(w.node)) { in DfsTraversal()
Dtopological_sort.cc82 for (int fanout : graph_view.GetFanout(ready_node)) { in ComputeTopologicalOrder()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dshape_optimizer.cc46 graph.GetFanout(MutableGraphView::OutputPort(&node, 0))) { in Optimize()
Dloop_optimizer.cc506 (!graph_view.GetFanout(fanout_idx).empty() || in GetStackPushNodesToConvert()
768 for (const MutableGraphView::InputPort& port : view.GetFanout(dead)) { in RemoveDeadBranches()
803 view.GetFanout(value_index); in RemoveDeadBranches()
Dmemory_optimizer.cc906 view.GetFanout(generator); in FindSwapOutTrigger()
1041 for (MutableGraphView::InputPort input : graph.GetFanout(port)) { in IdentifySwappingCandidates()
Dfunction_optimizer.cc1619 ctx->graph_view().GetFanout(control_output_port), in InlineIndirectFunctionCall()