Searched refs:ClusterOps (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_cluster_formation.cc | 67 using ClusterOps = llvm::SmallSetVector<Operation*, 8>; typedef
70 using ClusterMap = llvm::SmallDenseMap<llvm::StringRef, ClusterOps, 8>;
160 Block* block, Operation* op, const ClusterOps& cluster_ops, in ShouldMoveOpAfterCluster()
205 Block* block, const ClusterOps& cluster_ops, in CollectClusterPrecedingUsers()
234 Block* block, const ClusterOps& cluster_ops) { in CollectClusterResults()
254 Block* block, const ClusterOps& cluster_ops, llvm::ArrayRef<Value> results, in CreateClusterOp()
|