diff options
| author | Johan Alfven <johan.alfven@arm.com> | 2024-05-13 13:44:42 +0200 |
|---|---|---|
| committer | Johan Alfven <johan.alfven@arm.com> | 2024-05-15 14:51:05 +0200 |
| commit | 891468561ecfc61d27adcdc92b41ec216eaa1b08 (patch) | |
| tree | 5302e1e70549122d0c39c581a34db876010eb23f /ethosu/vela/tosa_reader.py | |
| parent | f49370003956d4f6f7d177114a68edb07b364fe9 (diff) | |
| download | ethos-u-vela-891468561ecfc61d27adcdc92b41ec216eaa1b08.tar.gz | |
MLBEDSW-9067: MLCE: Group Avgpool ops for concat
- Concat is implemented by several avgpool ops, all of them
writing to the same ofm but with a slice offset. If a compiled
network contains cpu fallbacks the avgpool ops might end up
running in different custom ops. This works fine as long as the
runtime provides the same scratch area. If not the output from
the concat might be corrupt.
- This fix adds an extra step to the pass packing so that all
avgpool ops for a concat is group together and run within the
same custom op in order to prevent possible corruption.
Change-Id: I343e08d7b4046f969b3d9ec3479db6490cbe4170
Signed-off-by: Johan Alfven <johan.alfven@arm.com>
Diffstat (limited to 'ethosu/vela/tosa_reader.py')
0 files changed, 0 insertions, 0 deletions