/external/tensorflow/tensorflow/core/grappler/ |
D | graph_topology_view_test.cc | 89 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()
|
D | graph_view.cc | 69 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()
|
D | mutable_graph_view.h | 51 using GraphViewInternal::GetFanout; 52 const absl::flat_hash_set<InputPort>& GetFanout(
|
D | graph_view_test.cc | 150 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()
|
D | graph_topology_view.h | 90 const absl::InlinedVector<int, 2>& GetFanout(int node_idx) const;
|
D | graph_topology_view.cc | 202 const absl::InlinedVector<int, 2>& GraphTopologyView::GetFanout( in GetFanout() function in tensorflow::grappler::GraphTopologyView
|
D | mutable_graph_view.cc | 417 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()
|
D | graph_view.h | 139 const absl::flat_hash_set<InputPort>& GetFanout( in GetFanout() function
|
D | mutable_graph_view_test.cc | 52 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/ |
D | latency_all_edges.cc | 85 auto fanout = graph.GetFanout(output_port); in OptimizeAndCollectStats()
|
/external/tensorflow/tensorflow/core/grappler/utils/ |
D | traversal.cc | 112 for (const int fanout : graph_view.GetFanout(w.node)) { in DfsTraversal()
|
D | topological_sort.cc | 82 for (int fanout : graph_view.GetFanout(ready_node)) { in ComputeTopologicalOrder()
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | shape_optimizer.cc | 46 graph.GetFanout(MutableGraphView::OutputPort(&node, 0))) { in Optimize()
|
D | loop_optimizer.cc | 506 (!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()
|
D | memory_optimizer.cc | 906 view.GetFanout(generator); in FindSwapOutTrigger() 1041 for (MutableGraphView::InputPort input : graph.GetFanout(port)) { in IdentifySwappingCandidates()
|
D | function_optimizer.cc | 1619 ctx->graph_view().GetFanout(control_output_port), in InlineIndirectFunctionCall()
|