| /external/tensorflow/tensorflow/core/grappler/ |
| D | mutable_graph_view.cc | 132 void SwapRegularFanoutInputs(FanoutsMap* fanouts, NodeDef* from_node, in SwapRegularFanoutInputs() argument
138 auto it = fanouts->find(port); in SwapRegularFanoutInputs()
139 if (it == fanouts->end()) { in SwapRegularFanoutInputs()
151 void SwapFanoutInputs(const MutableGraphView& graph, FanoutsMap* fanouts, in SwapFanoutInputs() argument
154 auto from_control_fanouts = fanouts->find({from_node, Graph::kControlSlot}); in SwapFanoutInputs()
155 if (from_control_fanouts != fanouts->end()) { in SwapFanoutInputs()
158 auto to_control_fanouts = fanouts->find({to_node, Graph::kControlSlot}); in SwapFanoutInputs()
159 if (to_control_fanouts != fanouts->end()) { in SwapFanoutInputs()
164 SwapRegularFanoutInputs(fanouts, from_node, to_node->name(), in SwapFanoutInputs()
169 SwapRegularFanoutInputs(fanouts, to_node, from_node->name(), in SwapFanoutInputs()
[all …]
|
| D | graph_view_test.cc | 164 absl::flat_hash_set<string> fanouts; in TEST_F() local
167 fanouts.insert(absl::StrCat(fo.node->name(), ":", fo.port_id)); in TEST_F()
170 EXPECT_EQ(fanouts, expected_fanouts); in TEST_F()
|
| D | mutable_graph_view_test.cc | 57 absl::Span<const string> fanouts) { in CompareNodeFanouts() argument
60 EXPECT_EQ(node_fanouts.size(), fanouts.size()); in CompareNodeFanouts()
61 for (const string& fanout : fanouts) { in CompareNodeFanouts()
73 absl::Span<const string> fanouts) { in CheckNode() argument
85 CompareNodeFanouts(graph, node, fanouts); in CheckNode()
1188 auto fanouts = graph.GetFanout(output_port); in CheckFanoutRemoved() local
1189 for (auto fanout : fanouts) { in CheckFanoutRemoved()
|
| D | graph_view.h | 360 absl::flat_hash_map<OutputPort, absl::flat_hash_set<InputPort>>& fanouts() { in fanouts() function
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | pattern_utils.h | 211 for (const auto& fanouts : fanouts_by_ports) { in HasRemoveNodeExternalDependents() local
212 for (const auto& fanout : fanouts) { in HasRemoveNodeExternalDependents()
|
| D | graph_view.cc | 856 void MutableGraphView::ReplaceNodeFanouts(MutableNodeView* node, T* fanouts) { in ReplaceNodeFanouts() argument
857 node->num_regular_fanouts_ = fanouts->num_regular_fanouts_; in ReplaceNodeFanouts()
858 node->regular_fanouts_by_port_ = std::move(fanouts->regular_fanouts_by_port_); in ReplaceNodeFanouts()
876 node->controlled_fanouts_ = std::move(fanouts->controlled_fanouts_); in ReplaceNodeFanouts()
918 auto& fanouts = fanouts_it->second; in FixRenamedNodes() local
919 ReplaceNodeFanouts(&renamed_node, &fanouts); in FixRenamedNodes()
974 NodeViewFanouts& fanouts = it->second; in AddNewNodes() local
975 ReplaceNodeFanouts(&new_node_view, &fanouts); in AddNewNodes()
1025 auto& fanouts = fanin_node_view->regular_fanouts_by_port_[fanin.index()]; in RemoveRegularFaninFanoutInternal() local
1026 int fanouts_size = fanouts.size(); in RemoveRegularFaninFanoutInternal()
[all …]
|
| D | graph_view.h | 480 void ReplaceNodeFanouts(MutableNodeView* node, T* fanouts);
|
| D | graph_view_test.cc | 461 absl::flat_hash_set<FaninViewType> fanouts(d_fanouts[i].begin(), in TYPED_TEST() local
463 EXPECT_TRUE(fanouts.contains( in TYPED_TEST()
465 EXPECT_TRUE(fanouts.contains( in TYPED_TEST()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | static_schedule.cc | 81 std::unordered_map<const NodeDef*, std::vector<const NodeDef*>> fanouts; in EstimateEarliestExecutionTimes() local
91 fanouts[fanin].push_back(&node); in EstimateEarliestExecutionTimes()
114 for (const NodeDef* fanout : fanouts[node]) { in EstimateEarliestExecutionTimes()
|
| D | common_subgraph_elimination.cc | 221 const auto fanouts = node_map.GetOutputs(node->name()); in DedupComputations() local
222 for (NodeDef* fanout : fanouts) { in DedupComputations()
|
| D | implementation_selector.cc | 355 for (const auto& fanouts : node_view->GetRegularFanouts()) { in SelectDeviceIndex() local
356 for (const auto& fanout : fanouts) { in SelectDeviceIndex()
|
| D | loop_optimizer.cc | 468 std::vector<int> fanouts; in GetStackPushNodesToConvert() local
475 fanouts.push_back(idx.value()); in GetStackPushNodesToConvert()
478 for (int fanout_idx : fanouts) { in GetStackPushNodesToConvert()
|
| D | constant_folding.cc | 512 auto fanouts = node_map_->GetOutputs(shape_n_node->name()); in MaterializeShapes() local
515 for (NodeDef* output : fanouts) { in MaterializeShapes()
2646 const auto& fanouts = node_map_->GetOutputs(node->name()); in SimplifySwitch() local
2647 if (fanouts.size() == 2) { in SimplifySwitch()
2648 for (const NodeDef* fanout : fanouts) { in SimplifySwitch()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | auto_shard.cc | 501 auto fanouts = graph->GetFanouts(*flat_map_node, false); in RecursivelyHandleOp() local
503 if (fanouts.size() == 1) { in RecursivelyHandleOp()
504 return ProcessDatasetSourceNode(graph, *fanouts.begin()->node, in RecursivelyHandleOp()
|