Home
last modified time | relevance | path

Searched refs:fanouts (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dmutable_graph_view.cc131 void SwapRegularFanoutInputs(FanoutsMap* fanouts, NodeDef* from_node, in SwapRegularFanoutInputs() argument
137 auto it = fanouts->find(port); in SwapRegularFanoutInputs()
138 if (it == fanouts->end()) { in SwapRegularFanoutInputs()
150 void SwapFanoutInputs(const MutableGraphView& graph, FanoutsMap* fanouts, in SwapFanoutInputs() argument
153 auto from_control_fanouts = fanouts->find({from_node, Graph::kControlSlot}); in SwapFanoutInputs()
154 if (from_control_fanouts != fanouts->end()) { in SwapFanoutInputs()
157 auto to_control_fanouts = fanouts->find({to_node, Graph::kControlSlot}); in SwapFanoutInputs()
158 if (to_control_fanouts != fanouts->end()) { in SwapFanoutInputs()
163 SwapRegularFanoutInputs(fanouts, from_node, to_node->name(), in SwapFanoutInputs()
168 SwapRegularFanoutInputs(fanouts, to_node, from_node->name(), in SwapFanoutInputs()
[all …]
Dgraph_view_test.cc163 absl::flat_hash_set<string> fanouts; in TEST_F() local
166 fanouts.insert(absl::StrCat(fo.node->name(), ":", fo.port_id)); in TEST_F()
169 EXPECT_EQ(fanouts, expected_fanouts); in TEST_F()
Dmutable_graph_view_test.cc57 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()
Dgraph_view.h360 absl::flat_hash_map<OutputPort, absl::flat_hash_set<InputPort>>& fanouts() { in fanouts() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dstatic_schedule.cc79 std::unordered_map<const NodeDef*, std::vector<const NodeDef*>> fanouts; in EstimateEarliestExecutionTimes() local
89 fanouts[fanin].push_back(&node); in EstimateEarliestExecutionTimes()
109 for (const NodeDef* fanout : fanouts[node]) { in EstimateEarliestExecutionTimes()
Dloop_optimizer.cc469 std::vector<int> fanouts; in GetStackPushNodesToConvert() local
476 fanouts.push_back(idx.value()); in GetStackPushNodesToConvert()
479 for (int fanout_idx : fanouts) { in GetStackPushNodesToConvert()
Dconstant_folding.cc2340 auto fanouts = node_map_->GetOutputs(node_name); in MoveConstantsPastEnter() local
2343 for (NodeDef* fanout : fanouts) { in MoveConstantsPastEnter()
2386 auto fanouts = node_map_->GetOutputs(node->name()); in SimplifySwitch() local
2387 if (fanouts.size() == 2) { in SimplifySwitch()
2388 for (NodeDef* fanout : fanouts) { in SimplifySwitch()
Darithmetic_optimizer.cc3480 std::vector<NodeDef*> fanouts(tmp.begin(), tmp.end()); in DedupComputations() local
3481 for (NodeDef* fanout : fanouts) { in DedupComputations()
/external/tensorflow/tensorflow/core/grappler/utils/
Dframe.cc52 absl::flat_hash_set<GraphView::InputPort> fanouts = in InferFromGraphView() local
55 for (const GraphView::InputPort& fanout : fanouts) { in InferFromGraphView()