| /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 | 151 EXPECT_EQ(graph.GetFanout(output).size(), 2); in TEST_F()
152 for (auto fanout : graph.GetFanout(output)) { in TEST_F()
195 auto fanout = graph.GetFanout(output); in TEST_F()
203 fanout = graph.GetFanout(output); in TEST_F()
397 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()
772 GetFanout(GraphView::OutputPort(from_node, Graph::kControlSlot)); in UpdateFanoutsInternal()
976 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/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 | 80 graph.GetFanout(MutableGraphView::OutputPort(&node, 0))) { in Optimize()
|
| D | loop_optimizer.cc | 505 (!graph_view.GetFanout(fanout_idx).empty() || in GetStackPushNodesToConvert()
796 for (const MutableGraphView::InputPort& port : view.GetFanout(dead)) { in RemoveDeadBranches()
831 view.GetFanout(value_index); in RemoveDeadBranches()
|
| D | auto_mixed_precision.cc | 438 const absl::InlinedVector<int, 2>& GetFanout(int node_idx) const;
663 const absl::InlinedVector<int, 2>& GraphTypeTopologyView::GetFanout( in GetFanout() function in tensorflow::grappler::__anon69ea7d4c0111::GraphTypeTopologyView
817 for (const int fanout : graph_type_view.GetFanout(w.node)) { in DfsTypeTraversal()
1732 const auto& fanout = graph_view_.GetFanout(output_port); in ForceColorMatchOnRecurrentEdges()
1859 const auto& fanout = graph_view_.GetFanout(src); in MakeCastsAllowIfAllOutputsAllow()
1922 auto fanout = graph_view_.GetFanout(src); in ChangeTypeAttrsAndAddCasts()
|
| D | memory_optimizer.cc | 937 view.GetFanout(generator); in FindSwapOutTrigger()
1076 for (MutableGraphView::InputPort input : graph.GetFanout(port)) { in IdentifySwappingCandidates()
|