Searched refs:nested_cluster (Results 1 – 1 of 1) sorted by relevance
80 op->walk([&](mlir::tf_device::ClusterOp nested_cluster) { in FindAllDeviceClusters() argument81 nested_clusters.emplace_back(nested_cluster); in FindAllDeviceClusters()362 for (mlir::tf_device::ClusterOp nested_cluster : nested_clusters) { in DecomposeIf() local363 if (mlir::failed(VerifyClusterInputOutput(nested_cluster))) in DecomposeIf()367 if (mlir::failed(ExtractMeshFromCluster(nested_cluster, &nested_mesh))) in DecomposeIf()378 if (IsInsideIfThenBranch(if_op, nested_cluster)) { in DecomposeIf()382 nested_cluster.GetBody().getOperations(); in DecomposeIf()391 nested_cluster.GetBody().getOperations(); in DecomposeIf()397 nested_cluster.erase(); in DecomposeIf()