Searched refs:cluster_names (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
| D | resolve_cluster.cc | 36 const std::vector<string>& cluster_names, GraphDef* graph) { in AddNodeToGraph() argument
46 for (const string& cluster_name : cluster_names) { in AddNodeToGraph()
98 std::vector<string> cluster_names; in MaybeResolveClusters() local
106 cluster_names.push_back((*itr)->GetName()); in MaybeResolveClusters()
117 AddNodeToGraph(*src_node, cluster_names, pruned_graph.get()); in MaybeResolveClusters()
125 AddNodeToGraph(node, cluster_names, pruned_graph.get()); in MaybeResolveClusters()
|
| D | resolve_svdf_test.cc | 165 std::vector<string> cluster_names; in TEST_F() local
170 cluster_names.push_back(cluster->GetName()); in TEST_F()
174 EXPECT_THAT(cluster_names, in TEST_F()
|
| D | resolve_cluster.h | 43 const std::vector<string>& cluster_names,
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_fusion_optimizer.cc | 324 std::unordered_map<int, string> cluster_names; in Optimize() local
333 string& name = cluster_names[cluster]; in Optimize()
|
| D | mark_for_compilation_pass_test.cc | 70 const Graph& g, std::vector<string>* cluster_names = nullptr) { in GetClusterSets() argument
71 CHECK(cluster_names == nullptr || cluster_names->empty()); in GetClusterSets()
77 if (cluster_names != nullptr) { in GetClusterSets()
78 cluster_names->push_back(p.first); in GetClusterSets()
82 if (cluster_names != nullptr) { in GetClusterSets()
83 std::sort(cluster_names->begin(), cluster_names->end()); in GetClusterSets()
604 std::vector<string> cluster_names; in TEST() local
606 GetClusterSets(*graph, &cluster_names); in TEST()
612 ASSERT_EQ(cluster_sets[cluster_names[0]], expected_clustered_nodes_a); in TEST()
|
| D | mark_for_compilation_pass.cc | 1287 std::unordered_map<int, string> cluster_names; in RunImpl() local
1333 string& name = cluster_names[cluster_repr]; in RunImpl()
|