| /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()
|