Searched defs:cluster_ops (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_cluster_formation.cc | 194 Operation* op, bool incoming, const OpSetVector& cluster_ops, in hasOpClusterControlDependency()
211 const OpSetVector& cluster_ops, in hasOpClusterDataDependency()
238 Block* block, const OpSetVector& cluster_ops, in CollectClusterSuccessorOps()
300 Block* block, const OpSetVector& cluster_ops) { in CollectClusterResults()
320 Block* block, const OpSetVector& cluster_ops, llvm::ArrayRef<Value> results, in CreateClusterOp()
593 const auto& cluster_ops = cluster_metadata_and_ops.getSecond(); in FormClustersInBlock() local
|
| D | tpu_extract_outside_compilation.cc | 280 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()
|
| D | tpu_host_computation_expansion.cc | 71 auto cluster_ops = cluster.GetBody().without_terminator(); in ExpandHeadOutsideCompiledOps() local
|
| D | tpu_extract_head_tail_outside_compilation.cc | 141 auto cluster_ops = cluster.GetBody().without_terminator(); in FindOutsideCompiledOpsAtHead() local
243 auto cluster_ops = llvm::reverse(cluster.GetBody().without_terminator()); in FindOutsideCompiledOpsAtTailAndClusterResults() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | designate_resource_handle_mesh.cc | 46 const auto& cluster_ops = cluster.GetBody().without_terminator(); in SetMeshForResourceCreatingCluster() local
|
| D | mesh_propagation.cc | 665 llvm::SmallVector<mlir::tf_device::ClusterOp, 8> cluster_ops; in PropagateMesh() local
|