Searched refs:dep_map (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.cc | 1033 const std::unordered_map<int, std::unordered_set<int>>& dep_map) in TransferParamsComparator() argument
1034 : dependency_map_(dep_map) {} in TransferParamsComparator()
1061 std::unordered_map<int, std::unordered_set<int>>& dep_map, in FillDependencyRec() argument
1063 if (dep_map.count(node_id) == 0 || dep_map.at(node_id).empty() || in FillDependencyRec()
1067 CHECK_EQ(dep_map.count(node_id), 1); in FillDependencyRec()
1070 for (int child_node_id : dep_map.at(node_id)) { in FillDependencyRec()
1075 FillDependencyRec(child_node_id, dep_map, completed); in FillDependencyRec()
1080 for (int child_node_id : dep_map.at(node_id)) { in FillDependencyRec()
1081 if (dep_map.count(child_node_id) == 0) { in FillDependencyRec()
1084 for (int depending_id : dep_map.at(child_node_id)) { in FillDependencyRec()
[all …]
|
| D | graph_transferer.h | 102 const std::unordered_map<int, std::unordered_set<int>>& dep_map);
205 int node_id, std::unordered_map<int, std::unordered_set<int>>& dep_map,
|