MLBEDSW-10826: Fix the axis attribute adjustment
This axis attribute refers to the IFM rank. There are ops where IFM rank is not equal to OFM rank, such as Reduce Min/Max with keep_dims = false. The correct way is to use IFM rank as the base for this adjustment.
Signed-off-by: Johan Gunnarsson johan.gunnarsson@arm.com Change-Id: I5aab7efef84b720fd15c8735f5c88df76ac24952