Searched refs:cluster_ops (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_cluster_formation.cc | 160 Block* block, Operation* op, const ClusterOps& cluster_ops, in ShouldMoveOpAfterCluster() argument
172 if (cluster_ops.count(def) != 0 || preceding_users.count(def) != 0) { in ShouldMoveOpAfterCluster()
205 Block* block, const ClusterOps& cluster_ops, in CollectClusterPrecedingUsers() argument
210 auto front = Block::iterator(cluster_ops.front()); in CollectClusterPrecedingUsers()
211 auto back = Block::iterator(cluster_ops.back()); in CollectClusterPrecedingUsers()
213 if (cluster_ops.contains(&op)) { in CollectClusterPrecedingUsers()
217 block, &op, cluster_ops, preceding_users, in CollectClusterPrecedingUsers()
234 Block* block, const ClusterOps& cluster_ops) { in CollectClusterResults() argument
237 for (Operation* op : cluster_ops) { in CollectClusterResults()
241 if (cluster_ops.count(block->findAncestorOpInBlock(*user)) == 0) { in CollectClusterResults()
[all …]
|
| D | tpu_host_computation_expansion.cc | 73 auto cluster_ops = cluster.GetBody().without_terminator(); in ExpandHeadOutsideCompiledOps() local
74 for (Operation& cluster_op : cluster_ops) { in ExpandHeadOutsideCompiledOps()
|
| D | tpu_extract_head_tail_outside_compilation.cc | 138 auto cluster_ops = cluster.GetBody().without_terminator(); in FindOutsideCompiledOpsAtHead() local
139 for (Operation& cluster_op : cluster_ops) { in FindOutsideCompiledOpsAtHead()
238 auto cluster_ops = llvm::reverse(cluster.GetBody().without_terminator()); in FindOutsideCompiledOpsAtTailAndClusterResults() local
239 for (Operation& cluster_op : cluster_ops) { in FindOutsideCompiledOpsAtTailAndClusterResults()
|