[Codegen] Allow memref type propagation through collapse_shape #15572

Job	Run time
setup / setup	7s
Build Packages / Linux Release (x86_64)	7m 45s
Unit Test / Linux (x86_64)	2m 44s
Test AMD MI300 / test_mi300	4m 18s
Regression Test / test_models :: cpu_llvm_task	4m 21s
Regression Test / test_regression_suite :: cpu_llvm_task	10m 11s
Regression Test / test_models :: amdgpu_rocm_mi250_gfx90a	9m 8s
Test AMD MI250 / test_mi250	6m 5s
Regression Test / test_regression_suite :: amdgpu_rocm_mi250_gfx90a	10m 29s
Test AMD W7900 / test_w7900	1m 49s
Regression Test / test_models :: amdgpu_rocm_mi300_gfx942	3m 27s
Regression Test / test_regression_suite :: amdgpu_rocm_mi300_gfx942	8m 35s
Regression Test / test_models :: amdgpu_vulkan	1m 35s
Test Android / android_arm64	5m 50s
Test ONNX / test_onnx_models :: cpu_llvm_task	2m 23s
Test PJRT plugin / Build and test (ubuntu-20.04, cpu)	3m 56s
Test ONNX / test_onnx_ops :: cpu_llvm_sync	4m 10s
Test ONNX / test_onnx_ops :: amdgpu_rocm_rdna3	8m 53s
Test RISC-V 64 / riscv64	5m 49s
Test ONNX / test_onnx_ops :: amdgpu_vulkan	1m 46s
Test Sharktank / test_sharktank_models :: cpu_llvm_task	43s
Test TensorFlow / Linux (x86_64)	1m 21s
	1h 45m 25s

Provide feedback