Arm backend: Add MM to BMM pass (#7341)
aten.mm does not support input or output tensors of rank 3, which is required by TOSA for the MM operation. Therefore, create a pass that converts any MM nodes to BMM (which always has rank 3). The pass also unsqueezes tensors of rank 2 to rank 3. As a result of the new pass, op_mm.py is no longer required and has been removed. Change-Id: I8459dd73bb366452b5139b48a5724c300b2d5a26 Signed-off-by: Sebastian Larsson <sebastian.larsson@arm.com>
S
Sebastian Larsson committed
05abf894f1c6dda34583fe715684b854a12ad5cc
Parent: b16271c
Committed by GitHub <noreply@github.com>
on 1/9/2025, 3:00:44 PM