Searched refs:is_node_in_cluster (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
| D | resolve_cluster.cc | 66 std::unordered_map<std::string, bool>* is_node_in_cluster, in FindCluster() argument
71 bool node_in_cluster = (*is_node_in_cluster)[node.name()]; in FindCluster()
79 (*is_node_in_cluster)[cluster_node->name()] = true; in FindCluster()
94 std::unordered_map<std::string, bool> is_node_in_cluster; in MaybeResolveClusters() local
96 is_node_in_cluster[node.name()] = false; in MaybeResolveClusters()
104 if (FindCluster(*cluster_factory, graph_def, &is_node_in_cluster, in MaybeResolveClusters()
124 bool node_in_cluster = is_node_in_cluster[node.name()]; in MaybeResolveClusters()
|
| D | resolve_svdf_test.cc | 160 std::unordered_map<std::string, bool> is_node_in_cluster; in TEST_F() local
162 is_node_in_cluster[node.name()] = false; in TEST_F()
166 CHECK(FindCluster(svdf_cluster_factory_, graph_, &is_node_in_cluster, in TEST_F()
|
| D | resolve_cluster.h | 52 std::unordered_map<std::string, bool>* is_node_in_cluster,
|