diff --git a/compilers/concrete-compiler/compiler/lib/Conversion/FHETensorOpsToLinalg/TensorOpsToLinalg.cpp b/compilers/concrete-compiler/compiler/lib/Conversion/FHETensorOpsToLinalg/TensorOpsToLinalg.cpp
index 25f6ba47da..9fdef0a734 100644
--- a/compilers/concrete-compiler/compiler/lib/Conversion/FHETensorOpsToLinalg/TensorOpsToLinalg.cpp
+++ b/compilers/concrete-compiler/compiler/lib/Conversion/FHETensorOpsToLinalg/TensorOpsToLinalg.cpp
@@ -2021,6 +2021,9 @@ struct TensorPartitionFrontierOpToLinalgGeneric
                                                  outs, maps, iteratorTypes, doc,
                                                  call, bodyBuilder);
 
+    if (pfOp->hasAttr("tile-sizes"))
+      genericOp->setAttr("tile-sizes", pfOp->getAttr("tile-sizes"));
+
     rewriter.replaceOp(pfOp, {genericOp.getResult(0)});
 
     return ::mlir::success();