Home
last modified time | relevance | path

Searched refs:cluster_names (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
Dresolve_cluster.cc36 const std::vector<std::string>& cluster_names, in AddNodeToGraph() argument
47 for (const std::string& cluster_name : cluster_names) { in AddNodeToGraph()
99 std::vector<std::string> cluster_names; in MaybeResolveClusters() local
107 cluster_names.push_back((*itr)->GetName()); in MaybeResolveClusters()
118 AddNodeToGraph(*src_node, cluster_names, pruned_graph.get()); in MaybeResolveClusters()
126 AddNodeToGraph(node, cluster_names, pruned_graph.get()); in MaybeResolveClusters()
Dresolve_svdf_test.cc165 std::vector<std::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()
Dresolve_cluster.h43 const std::vector<std::string>& cluster_names,
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass_test.cc75 const Graph& g, std::vector<string>* cluster_names = nullptr) { in GetClusterSets() argument
76 CHECK(cluster_names == nullptr || cluster_names->empty()); in GetClusterSets()
82 if (cluster_names != nullptr) { in GetClusterSets()
83 cluster_names->push_back(p.first); in GetClusterSets()
87 if (cluster_names != nullptr) { in GetClusterSets()
88 std::sort(cluster_names->begin(), cluster_names->end()); in GetClusterSets()
737 std::vector<string> cluster_names; in TEST() local
739 GetClusterSets(*graph, &cluster_names); in TEST()
745 ASSERT_EQ(cluster_sets[cluster_names[0]], expected_clustered_nodes_a); in TEST()
Dmark_for_compilation_pass.cc852 std::unordered_map<int, string> cluster_names; in CreateClusters() local
880 string& name = cluster_names[cluster->cycles_graph_node_id()]; in CreateClusters()
/external/mesa3d/src/freedreno/decode/
Dcrashdec.c764 const char *cluster_names[6] = { in dump_cp_mem_pool() local
767 printf("\tCLUSTER_%s:\n\n", cluster_names[queue]); in dump_cp_mem_pool()
/external/rust/crates/grpcio-sys/grpc/src/core/ext/xds/
Dxds_api.h455 std::set<std::string>* cluster_names,
Dxds_api.cc2270 std::set<std::string>* cluster_names, in ParseLrsResponse() argument
2293 cluster_names->emplace(UpbStringToStdString(clusters[i])); in ParseLrsResponse()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc1613 std::unordered_set<string> cluster_names; in RenameClustersWithDuplicatedNames() local
1621 cluster_names.insert(cluster_name); in RenameClustersWithDuplicatedNames()
1636 if (cluster_names.find(new_cluster_name) == cluster_names.end()) { in RenameClustersWithDuplicatedNames()
1641 cluster_names.insert(new_cluster_name); in RenameClustersWithDuplicatedNames()