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

MLBEDSW-6941: Set correct OFM shape for fc op



If IFM operator shape is rewritten so that batching
is greater than one for fully connect, the OFM batch
must also be calculated. This change will fix output diffs
for networks that have fully connect OFM with rank greater
than 2.

Signed-off-by: Johan Alfvén's avatarJohan Alfven <johan.alfven@arm.com>
Change-Id: I5009edc647a1449a02c8116b45808c1c68beffe6
parent a64616c4
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