Propagate expand_shape past collapse_shape

Signed-off-by: MaheshRavishankar <mravisha@smc300x-clt-r4c4-02.cs-clt.dcgpu>

compiler/src/iree/compiler/DispatchCreation/BubbleUpExpandShapes.cpp

@@ -204,6 +204,7 @@ void BubbleUpExpandShapesPass::runOnOperation() {
   // that can be done later) of reshape ops.
   tensor::populateFoldTensorEmptyPatterns(bubbleExpandShapePatterns);
   bubbleExpandShapePatterns.insert<BubbleExpandThroughExtract>(context);
+  tensor::populateBubbleUpExpandShapePatterns(bubbleExpandShapePatterns);
   tensor::ExpandShapeOp::getCanonicalizationPatterns(bubbleExpandShapePatterns,
                                                      context);