Searched refs:fanouts (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/ |
D | mutable_graph_view.cc | 131 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 …]
|
D | graph_view_test.cc | 163 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()
|
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/optimizers/ |
D | static_schedule.cc | 79 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()
|
D | loop_optimizer.cc | 469 std::vector<int> fanouts; in GetStackPushNodesToConvert() local 476 fanouts.push_back(idx.value()); in GetStackPushNodesToConvert() 479 for (int fanout_idx : fanouts) { in GetStackPushNodesToConvert()
|
D | constant_folding.cc | 2340 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()
|
D | arithmetic_optimizer.cc | 3480 std::vector<NodeDef*> fanouts(tmp.begin(), tmp.end()); in DedupComputations() local 3481 for (NodeDef* fanout : fanouts) { in DedupComputations()
|
/external/tensorflow/tensorflow/core/grappler/utils/ |
D | frame.cc | 52 absl::flat_hash_set<GraphView::InputPort> fanouts = in InferFromGraphView() local 55 for (const GraphView::InputPort& fanout : fanouts) { in InferFromGraphView()
|