Stephan Herhut a43f7d6d76 [mlir][tensor] Extend reshape utils. ...

This change changes the handling of trailing dimensions with unknown
extent. Users of the changessociationIndicesForReshape helper should
see benefits when transforming reshape like operations into
expand/collapse pairs if the higher-rank type has trailing unknown
dimensions.

The motivating example is a reshape from tensor<16x1x?xi32> to
tensor<16xi32> that can be modeled as collapsing the three dimensions.

Differential Revision: https://reviews.llvm.org/D119730

2022-02-18 09:57:39 +01:00

9.9 KiB

Raw Blame History

View Raw