Searched refs:GetFanins (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | pin_to_host_optimizer.cc | 125 for (const auto& fanin : graph.GetFanins(node, false)) { in IsNodeOutputPortHostFriendly()
245 graph.GetFanins(node, /*include_controlling_nodes=*/false)) { in IsNodeHostCandidate()
|
| D | model_pruner.cc | 40 graph_view.GetFanins(node, /*include_controlling_nodes=*/true)) { in IsTrivialIdentity()
120 graph_view.GetFanins(node, /*include_controlling_nodes=*/true)) { in CanRemoveNode()
|
| D | memory_optimizer.cc | 529 for (const auto& input : view.GetFanins(node, false)) { in SchedulingPass()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | auto_shard.cc | 321 for (const auto& fanin : graph->GetFanins(node, true)) { in RemoveShuffleDataset()
343 for (const auto& fanin : graph->GetFanins(node, true)) { in RemoveShuffleDatasetV2()
370 for (const auto& fanin : graph->GetFanins(node, true)) { in RemoveShuffleDatasetV3()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | graph_view_test.cc | 174 for (const auto& fi : graph.GetFanins(*add_node, false)) { in TEST_F()
465 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()
|