Searched refs:GetFanin (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/ |
| D | graph_topology_view_test.cc | 83 EXPECT_EQ(graph_view.GetFanin(0), Fanin()); in TEST_F()
84 EXPECT_EQ(graph_view.GetFanin(1), Fanin()); in TEST_F()
85 EXPECT_EQ(graph_view.GetFanin(2), Fanin({0, 1})); in TEST_F()
86 EXPECT_EQ(graph_view.GetFanin(3), Fanin({0, 2})); in TEST_F()
108 EXPECT_EQ(graph_view.GetFanin(2), Fanin({0, 1, 3})); in TEST_F()
109 EXPECT_EQ(graph_view.GetFanin(3), Fanin({0, 2})); in TEST_F()
130 EXPECT_EQ(graph_view.GetFanin(2), Fanin({0, 1, 3})); in TEST_F()
131 EXPECT_EQ(graph_view.GetFanin(3), Fanin({0, 2})); in TEST_F()
143 EXPECT_EQ(graph_view.GetFanin(2), Fanin({0, 1})); in TEST_F()
144 EXPECT_EQ(graph_view.GetFanin(3), Fanin({0, 2})); in TEST_F()
|
| D | mutable_graph_view.h | 55 using GraphViewInternal::GetFanin;
56 absl::flat_hash_set<OutputPort> GetFanin(
|
| D | graph_view_test.cc | 210 auto fanin = graph.GetFanin(input); in TEST_F()
218 fanin = graph.GetFanin(input); in TEST_F()
226 fanin = graph.GetFanin(input); in TEST_F()
412 graph_view.GetFanin({node, 0}); in BM_GraphViewGetFanin()
|
| D | graph_topology_view.h | 87 const absl::InlinedVector<int, 4>& GetFanin(int node_idx) const;
|
| D | graph_view.cc | 79 return !graph_view.GetFanin(control_port).empty(); in HasControlFanin()
|
| D | graph_topology_view.cc | 194 const absl::InlinedVector<int, 4>& GraphTopologyView::GetFanin( in GetFanin() function in tensorflow::grappler::GraphTopologyView
|
| D | graph_view.h | 145 absl::flat_hash_set<OutputPort> GetFanin(const InputPort& port) const { in GetFanin() function
|
| D | mutable_graph_view.cc | 422 absl::flat_hash_set<MutableGraphView::OutputPort> MutableGraphView::GetFanin( in GetFanin() function in tensorflow::grappler::MutableGraphView
424 return GetFanin(MutableGraphView::InputPort(const_cast<NodeDef*>(port.node), in GetFanin()
|
| D | mutable_graph_view_test.cc | 51 EXPECT_TRUE(graph.GetFanin(input_port).contains(output_port)); in CompareNodeFanins()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | topological_sort.cc | 67 if (graph_view.GetFanin(i).empty()) { in ComputeTopologicalOrder()
72 for (int input : graph_view.GetFanin(i)) { in ComputeTopologicalOrder()
84 if (num_ready_inputs[fanout] == graph_view.GetFanin(fanout).size()) { in ComputeTopologicalOrder()
|
| D | traversal.cc | 108 for (const int fanin : graph_view.GetFanin(w.node)) { in DfsTraversal()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 1609 ctx->graph_view().GetFanin(control_input_port), in InlineIndirectFunctionCall()
|