diff --git a/compiler/src/iree/compiler/InputConversion/StableHLO/StableHLOToIREEInputDialects.cpp b/compiler/src/iree/compiler/InputConversion/StableHLO/StableHLOToIREEInputDialects.cpp
index 00dfbfdb2bf3..9fb5dc1f6ed6 100644
--- a/compiler/src/iree/compiler/InputConversion/StableHLO/StableHLOToIREEInputDialects.cpp
+++ b/compiler/src/iree/compiler/InputConversion/StableHLO/StableHLOToIREEInputDialects.cpp
@@ -529,6 +529,8 @@ struct ConvertStableHloToIreeInputDialects final
           ml_program::GlobalLoadConstOp::getOperationName(),
           ml_program::GlobalStoreOp::getOperationName(),
           tensor::FromElementsOp::getOperationName(),
+          tensor::CollapseShapeOp::getOperationName(),
+          tensor::ExpandShapeOp::getOperationName(),
           arith::IndexCastUIOp::getOperationName()}) {
       patterns.add<GenericTypeConvert>(opName, *typeConverter, context);
     }