MLBEDSW-9536: Add support for transpose with large axis
* Use DecomposeLargeAxis, but with untransposed OFM shape (ie. the
IFM shape). Transpose the OFM offset and shape in
DecomposeLargeAxis.
* Don't pad shapes in RearrangeTranspose because then they won't
match the perm attribute.
* At the end of decomposition, there can now be multiple ops for
each swap, so we must look for all ops using the first and last
tensor when adjusting the tensor and quantization.
Signed-off-by:
Johan Gunnarsson <johan.gunnarsson@arm.com>
Change-Id: I3bf3579c247faef42ddd83ae09b803a4de35910d
Loading
Please register or sign in to comment