Check >=2 dyn for source and result

Signed-off-by: Ian Wood <[email protected]>

Author: IanWood1

mlir/include/mlir/Dialect/Utils/ReshapeOpsUtils.h

@@ -387,7 +387,8 @@ struct ComposeExpandOfCollapseOp : public OpRewritePattern<ExpandOpTy> {
       auto resultSubShape =
           resultShape.slice(resultIndices.front(), resultIndices.size());
 
-      if (llvm::count_if(srcSubShape, ShapedType::isDynamic) >= 2)
+      if (llvm::count_if(srcSubShape, ShapedType::isDynamic) >= 2 &&
+          llvm::count_if(resultSubShape, ShapedType::isDynamic) >= 2)
         return std::nullopt;
 
       if (srcSubShape.size() == resultSubShape.size()) {