tree 974c41393b836e7aeef9b1c1dda2faef3f63ac94
parent 1d6d5c47c2000facc377620a64084738339ccda9
author Patrik Gustavsson <patrik.gustavsson@arm.com> 1618415650 +0200
committer Patrik Gustavsson <patrik.gustavsson@arm.com> 1618492744 +0200

MLBEDSW-4397 Fix Reshape ifm/ofm prod/cons by cpu op

Not only the sg input outputs need to be considered
before removing Reshape.

Added check if Reshape ifm/ofm is produced respectively
consumed by CPU. Handling is the same as if tensor is
sg input/output.

Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: If509e1d23e3f22ed4c963d8dabd8c00c6b9c07e3