Home
last modified time | relevance | path

Searched refs:src_cluster (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dmerge_clusters.cc120 mlir::tf_device::ClusterOp src_cluster, in MergeClusterMetadata() argument
122 if (mlir::failed(ValidateMetadataAttributes(src_cluster)) || in MergeClusterMetadata()
130 src_cluster->getAttrOfType<mlir::DenseIntElementsAttr>( in MergeClusterMetadata()
133 src_cluster->getAttrOfType<mlir::ArrayAttr>(kNewResourceArgLayouts); in MergeClusterMetadata()
159 src_cluster, src_resource_handle_indices_metadata, in MergeClusterMetadata()
178 src_cluster->getAttrOfType<mlir::ArrayAttr>(kShapeOpInputLayout); in MergeClusterMetadata()
180 src_cluster->getAttrOfType<mlir::DenseIntElementsAttr>( in MergeClusterMetadata()
203 src_cluster, src_shape_op_indices, src_shape_layouts, in MergeClusterMetadata()
H A Ddevice_mesh_cluster_coarsening.cc76 void MoveOpsInsideCluster(mlir::tf_device::ClusterOp src_cluster, in MoveOpsInsideCluster() argument
79 auto& cluster_body = src_cluster.GetBody().getOperations(); in MoveOpsInsideCluster()
H A Dhandle_cross_cluster_dependencies.cc184 auto src_cluster = in LowerToSendRecv() local
187 src_cluster.GetBody().getTerminator()->getOperand(result_index); in LowerToSendRecv()