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