Searched refs:ExtractMeshFromCluster (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | handle_cross_cluster_dependencies.cc | 55 mlir::LogicalResult ExtractMeshFromCluster(mlir::tf_device::ClusterOp cluster, in ExtractMeshFromCluster() function
142 if (mlir::failed(ExtractMeshFromCluster(cluster, &mesh))) in CloneConstantsAcrossMesh()
158 if (mlir::failed(ExtractMeshFromCluster(input_cluster, &input_mesh))) { in CloneConstantsAcrossMesh()
270 if (mlir::failed(ExtractMeshFromCluster(cluster, ¤t_mesh))) in ReplaceCopyToMeshWithVirtualSendRecv()
289 if (mlir::failed(ExtractMeshFromCluster(input_cluster, &input_mesh))) { in ReplaceCopyToMeshWithVirtualSendRecv()
|
| D | lower_send_recv.cc | 49 mlir::LogicalResult ExtractMeshFromCluster(mlir::tf_device::ClusterOp cluster, in ExtractMeshFromCluster() function
84 if (mlir::failed(ExtractMeshFromCluster( in LowerDTensorSendRecvsOps()
|
| D | merge_clusters.cc | 63 mlir::LogicalResult ExtractMeshFromCluster(mlir::tf_device::ClusterOp cluster, in ExtractMeshFromCluster() function
231 if (mlir::failed(ExtractMeshFromCluster(cluster, &cluster_mesh))) in InlineNestedDeviceClusters()
236 ExtractMeshFromCluster(parent_cluster, &parent_cluster_mesh))) in InlineNestedDeviceClusters()
367 if (mlir::failed(ExtractMeshFromCluster(nested_cluster, &nested_mesh))) in DecomposeIf()
491 if (mlir::failed(ExtractMeshFromCluster(cluster, &mesh))) in MergeClusters()
|