Lines Matching defs:mesh

25 using namespace mlir::mesh;
39     mesh::MeshOp mesh = symbolTable.lookupNearestSymbolFrom<mesh::MeshOp>(
46             mesh == symbolTable.lookupNearestSymbolFrom<mesh::MeshOp>(
64           symbolTable.lookupNearestSymbolFrom<mesh::MeshOp>(
67                   .getMeshAttr()) != mesh) {
73           shardShapedType(op.getResult().getType(), mesh, op.getSharding());
79           reshard(builder, mesh, op, targetShardOp, sourceShard);
110     return "test-mesh-resharding-spmdization";