Home
last modified time | relevance | path

Searched defs:cluster_mesh (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dmesh_propagation.cc556 auto cluster_mesh = cluster->getAttrOfType<mlir::StringAttr>(kMeshAttr); in PropagateMeshFromInputs() local
615 auto cluster_mesh = cluster->getAttrOfType<mlir::StringAttr>(kMeshAttr); in PropagateMeshFromConsumers() local
H A Dmerge_clusters.cc230 Mesh cluster_mesh; in InlineNestedDeviceClusters() local