Skip to content
Commit 968818d2 authored by Johan Gunnarsson's avatar Johan Gunnarsson
Browse files

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's avatarJohan Gunnarsson <johan.gunnarsson@arm.com>
Change-Id: I3bf3579c247faef42ddd83ae09b803a4de35910d
parent bd9820da
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment