SIGN IN SIGN UP

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