Commit 538ce0a
committed
Arm backend: Add dim mapping helpers
Dim args such as sum(dim=1) needs to be transformed
when swapping place with operators which change
shape, i.e. permutes and views.
ViewMap and PermuteMap handles and validates these
transforms for reduction dims and permute dims.
Signed-off-by: Adrian Lundell <adrian.lundell@arm.com>
Change-Id: I634f4494df37294d4ac3397f8457bedfd20f58301 parent 7d365ec commit 538ce0a
2 files changed
Lines changed: 1264 additions & 0 deletions
0 commit comments