Home
last modified time | relevance | path

Searched defs:cluster_ops (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_cluster_formation.cc161 Block* block, Operation* op, const ClusterOps& cluster_ops, in ShouldMoveOpAfterCluster()
206 Block* block, const ClusterOps& cluster_ops, in CollectClusterPrecedingUsers()
235 Block* block, const ClusterOps& cluster_ops) { in CollectClusterResults()
255 Block* block, const ClusterOps& cluster_ops, llvm::ArrayRef<Value> results, in CreateClusterOp()
558 const auto& cluster_ops = cluster_metadata_and_ops.getSecond(); in FormClustersInBlock() local
Dtpu_extract_outside_compilation.cc280 bool HasDynamicOutputs(const llvm::SmallSetVector<Operation*, 4>& cluster_ops) { in HasDynamicOutputs()
310 const llvm::SmallSetVector<Operation*, 4>& cluster_ops) { in GetStaticExternalOperands()
340 const llvm::SmallSetVector<Operation*, 4>& cluster_ops) { in GetAllExternalOperands()
359 const llvm::SmallSetVector<Operation*, 4>& cluster_ops) { in GetExternalOperands()
372 const llvm::SmallSetVector<Operation*, 4>& cluster_ops) { in GetExternalOutputs()
Dtpu_host_computation_expansion.cc71 auto cluster_ops = cluster.GetBody().without_terminator(); in ExpandHeadOutsideCompiledOps() local
Dtpu_extract_head_tail_outside_compilation.cc139 auto cluster_ops = cluster.GetBody().without_terminator(); in FindOutsideCompiledOpsAtHead() local
241 auto cluster_ops = llvm::reverse(cluster.GetBody().without_terminator()); in FindOutsideCompiledOpsAtTailAndClusterResults() local