dcaballe wrote: > I also see there are similar patterns Could we try to refactor that code into a utility? The only difference should be the resulting shape but the decomposition code should be the same? https://github.com/llvm/llvm-project/pull/164010