Searched refs:preceding_users (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_cluster_formation.cc | 161 const llvm::SmallSetVector<Operation*, 8>& preceding_users, in ShouldMoveOpAfterCluster() argument
172 if (cluster_ops.count(def) != 0 || preceding_users.count(def) != 0) { in ShouldMoveOpAfterCluster()
207 llvm::SmallSetVector<Operation*, 8> preceding_users; in CollectClusterPrecedingUsers() local
217 block, &op, cluster_ops, preceding_users, in CollectClusterPrecedingUsers()
219 preceding_users.insert(&op); in CollectClusterPrecedingUsers()
225 return preceding_users; in CollectClusterPrecedingUsers()
255 llvm::ArrayRef<Operation*> preceding_users) { in CreateClusterOp() argument
297 for (Operation* user : preceding_users) user->moveBefore(op_after_cluster); in CreateClusterOp()
553 llvm::SmallSetVector<Operation*, 8> preceding_users = in FormClustersInBlock() local
561 block, cluster_ops, results, preceding_users.getArrayRef()); in FormClustersInBlock()
|