Home
last modified time | relevance | path

Searched refs:fanouts_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dgraph_view.h141 return gtl::FindWithDefault(fanouts_, port, fanout_not_found_value_); in GetFanout()
218 auto it = fanouts_.find(port); in GetFanouts()
219 if (it != fanouts_.end()) { in GetFanouts()
266 auto it = fanouts_.find(port); in NumFanouts()
267 if (it != fanouts_.end()) count += it->second.size(); in NumFanouts()
287 auto it = fanouts_.find(port); in GetFanoutEdges()
288 if (it != fanouts_.end()) { in GetFanoutEdges()
344 fanouts_[output].emplace(node, -1); in AddFanouts()
349 fanouts_[output].emplace(node, i); in AddFanouts()
361 return fanouts_; in fanouts()
[all …]
Dgraph_topology_view.cc54 fanouts_.resize(num_nodes_); in InitializeFromGraph()
97 fanouts_[src_idx].push_back(dst_idx); in InitializeFromGraph()
127 fanouts_[input_idx].push_back(node_idx); in InitializeFromGraph()
137 SortAndRemoveDuplicates(&fanouts_[node_idx]); in InitializeFromGraph()
207 return is_valid_node_idx ? fanouts_[node_idx] : empty_fanout_; in GetFanout()
Dgraph_topology_view.h104 std::vector<absl::InlinedVector<int, 2>> fanouts_; // node_idx->output nodes variable
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision.cc467 std::vector<absl::InlinedVector<int, 2>> fanouts_; member in tensorflow::grappler::__anonab6217530111::GraphTypeTopologyView
508 fanouts_.resize(num_nodes_); in InitializeFromGraph()
549 fanouts_[input_node_type_idx].push_back(node_type_idx); in InitializeFromGraph()
559 SortAndRemoveDuplicates(&fanouts_[node_type_idx]); in InitializeFromGraph()
602 fanouts_[src_node_type_idx].push_back(dst_node_type_idx); in AddEphemeralEdges()
609 SortAndRemoveDuplicates(&fanouts_[node_type_idx]); in AddEphemeralEdges()
668 return is_valid_node_idx ? fanouts_[node_idx] : empty_fanout_; in GetFanout()