tree f434596481b21514e43d687c74bf94481caf71c1
parent d0e41cf4c20a0f5780a8dd072df29bdb3267d960
author Johan Alfven <johan.alfven@arm.com> 1676384403 +0100
committer tim.hall <tim.hall@arm.com> 1676486128 +0000

MLBEDSW-7347: MLCE: Split followed by elementwise op will assert

- The problem was that when the split slice read was moved
to the tensor consumer, in this case an elementwise operator,
this was not taken into account when the npu op for the
element wise operator was created. The npu op was created
with wrong ifm_width and ifm and ifm2 ended up with different
sizes. As a result, broadcasting is expected but that is
not True so the assert was triggered.
- The fix is to use the ifm box in order to set the correct
ifm_width for the npu operator.

Change-Id: I3291d34e7f8e7add9caf2296cca600c60e96bf7e
Signed-off-by: Johan Alfven <johan.alfven@arm.com>