Skip to content
Commit e4a98830 authored by Johan Alfvén's avatar Johan Alfvén
Browse files

MLBEDSW-9133: Output diff on MEAN op



 - A MEAN op with IFM rank two and reduce in C dimension caused an
output diff
 - The reason was that the intermediate tensor for calculating the sum
had the wrong shape because the reduceAxis shape had been padded to 4D
 - The fix is to use the original reduceAxis shape when calculating the
shape for the sum tensor

Change-Id: I144adfe07f697fecba6f7237e8b216295654f8ae
Signed-off-by: Johan Alfvén's avatarJohan Alfven <johan.alfven@arm.com>
parent 3ce02f61
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment