Searched refs:cluster_deps (Results 1 – 2 of 2) sorted by relevance
349 auto cluster_deps = absl::make_unique< in OutsideCompilationClusterDependencies() local360 auto dst_deps_it = cluster_deps->find(*dst_outside_compilation); in OutsideCompilationClusterDependencies()361 if (dst_deps_it == cluster_deps->end()) { in OutsideCompilationClusterDependencies()362 cluster_deps->insert(std::make_pair( in OutsideCompilationClusterDependencies()374 for (auto it = cluster_deps->begin(); it != cluster_deps->end(); it++) { in OutsideCompilationClusterDependencies()
312 const absl::flat_hash_map<string, std::vector<string>>& cluster_deps) { in BuildXlaHostComputeNodeDef() argument348 auto cluster_deps_it = cluster_deps.find(original_oc_name); in BuildXlaHostComputeNodeDef()349 if (cluster_deps_it != cluster_deps.end()) { in BuildXlaHostComputeNodeDef()382 const absl::flat_hash_map<string, std::vector<string>>& cluster_deps) { in ReplaceOutsideCompilationCallNode() argument386 BuildXlaHostComputeNodeDef(call_node, host_compute_core, cluster_deps)); in ReplaceOutsideCompilationCallNode()2354 TF_ASSIGN_OR_RETURN(auto cluster_deps, in ExtractOutsideCompilationForFunction()2408 graph_out.get(), n, host_compute_core, *cluster_deps); in ExtractOutsideCompilationForFunction()