Searched refs:cluster_to_tpu_ops_map (Results 1 – 1 of 1) sorted by relevance
287 cluster_to_tpu_ops_map, in MergeIsland()313 if (cluster_to_tpu_ops_map.count(cluster_name)) { in MergeIsland()314 for (auto tpu_op : cluster_to_tpu_ops_map[cluster_name]) { in MergeIsland()565 cluster_to_tpu_ops_map, in ExcludeUnqualifiedIdentityOp()567 for (auto& [target_cluster_name, tpu_ops] : cluster_to_tpu_ops_map) { in ExcludeUnqualifiedIdentityOp()577 cluster_to_tpu_ops_map) { in ExcludeUnqualifiedIdentityOp()578 ExcludeUnqualifiedIdentityOp(cluster_to_tpu_ops_map, /*incoming=*/true); in ExcludeUnqualifiedIdentityOp()579 ExcludeUnqualifiedIdentityOp(cluster_to_tpu_ops_map, /*incoming=*/false); in ExcludeUnqualifiedIdentityOp()650 cluster_to_tpu_ops_map; in runOnOperation() local654 cluster_to_tpu_ops_map, visited_ops))) { in runOnOperation()[all …]