Searched refs:GetFanins (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | pin_to_host_optimizer.cc | 121 for (const auto& fanin : graph.GetFanins(node, false)) { in IsNodeOutputPortHostFriendly()
241 graph.GetFanins(node, /*include_controlling_nodes=*/false)) { in IsNodeHostCandidate()
|
| D | model_pruner.cc | 39 graph_view.GetFanins(node, /*include_controlling_nodes=*/true)) { in IsTrivialIdentity()
102 graph_view.GetFanins(node, /*include_controlling_nodes=*/true)) { in CanRemoveNode()
|
| D | memory_optimizer.cc | 522 for (const auto& input : view.GetFanins(node, false)) { in SchedulingPass()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | auto_shard.cc | 167 for (const auto& fanin : graph->GetFanins(node, true)) { in RemoveShuffleDataset()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | graph_view_test.cc | 173 for (const auto& fi : graph.GetFanins(*add_node, false)) { in TEST_F()
461 graph_view.GetFanins(*node, /*include_controlling_nodes=*/false); in BM_GraphViewGetFanins()
|
| D | graph_view.h | 228 absl::flat_hash_set<OutputPort> GetFanins( in GetFanins() function
|
| D | mutable_graph_view_test.cc | 111 mutable_graph.GetFanins(node, /*include_controlling_nodes=*/true); in CheckGraph()
113 immutable_graph.GetFanins(node, /*include_controlling_nodes=*/true); in CheckGraph()
|