Home
last modified time | relevance | path

Searched refs:cluster_ops (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_cluster_formation.cc194 Operation* op, bool incoming, const OpSetVector& cluster_ops, in hasOpClusterControlDependency() argument
198 return cluster_ops.contains(other_op) || in hasOpClusterControlDependency()
211 const OpSetVector& cluster_ops, in hasOpClusterDataDependency() argument
224 if (cluster_ops.contains(candidate_op) || in hasOpClusterDataDependency()
238 Block* block, const OpSetVector& cluster_ops, in CollectClusterSuccessorOps() argument
247 auto rfront = Block::reverse_iterator(cluster_ops.front()); in CollectClusterSuccessorOps()
248 auto rback = Block::reverse_iterator(cluster_ops.back()); in CollectClusterSuccessorOps()
250 if (cluster_ops.contains(&op)) continue; in CollectClusterSuccessorOps()
252 hasOpClusterDataDependency(&op, /*incoming=*/false, cluster_ops, in CollectClusterSuccessorOps()
254 hasOpClusterControlDependency(&op, /*incoming=*/false, cluster_ops, in CollectClusterSuccessorOps()
[all …]
Dtpu_extract_outside_compilation.cc280 bool HasDynamicOutputs(const llvm::SmallSetVector<Operation*, 4>& cluster_ops) { in HasDynamicOutputs() argument
281 for (Operation* op : cluster_ops) { in HasDynamicOutputs()
283 if (cluster_ops.count(use.getOwner())) { in HasDynamicOutputs()
310 const llvm::SmallSetVector<Operation*, 4>& cluster_ops) { in GetStaticExternalOperands() argument
312 for (Operation* op : cluster_ops) { in GetStaticExternalOperands()
340 const llvm::SmallSetVector<Operation*, 4>& cluster_ops) { in GetAllExternalOperands() argument
342 for (Operation* op : cluster_ops) { in GetAllExternalOperands()
346 if (!defining_op || !cluster_ops.count(defining_op)) { in GetAllExternalOperands()
359 const llvm::SmallSetVector<Operation*, 4>& cluster_ops) { in GetExternalOperands() argument
362 bool has_dynamic_outputs = HasDynamicOutputs(cluster_ops); in GetExternalOperands()
[all …]
Dtpu_host_computation_expansion.cc71 auto cluster_ops = cluster.GetBody().without_terminator(); in ExpandHeadOutsideCompiledOps() local
72 for (Operation& cluster_op : cluster_ops) { in ExpandHeadOutsideCompiledOps()
Dtpu_extract_head_tail_outside_compilation.cc141 auto cluster_ops = cluster.GetBody().without_terminator(); in FindOutsideCompiledOpsAtHead() local
142 for (Operation& cluster_op : cluster_ops) { in FindOutsideCompiledOpsAtHead()
243 auto cluster_ops = llvm::reverse(cluster.GetBody().without_terminator()); in FindOutsideCompiledOpsAtTailAndClusterResults() local
244 for (Operation& cluster_op : cluster_ops) { in FindOutsideCompiledOpsAtTailAndClusterResults()
/external/tensorflow/tensorflow/dtensor/mlir/
Ddesignate_resource_handle_mesh.cc46 const auto& cluster_ops = cluster.GetBody().without_terminator(); in SetMeshForResourceCreatingCluster() local
49 llvm::count_if(cluster_ops, [](auto& operation) { in SetMeshForResourceCreatingCluster()
Dmesh_propagation.cc665 llvm::SmallVector<mlir::tf_device::ClusterOp, 8> cluster_ops; in PropagateMesh() local
667 cluster_ops.emplace_back(cluster); in PropagateMesh()
676 for (auto cluster : llvm::reverse(cluster_ops)) { in PropagateMesh()