diff options
author | Diqing Zhong <diqing.zhong@arm.com> | 2022-03-09 12:23:47 +0100 |
---|---|---|
committer | Diqing Zhong <diqing.zhong@arm.com> | 2022-03-11 17:08:15 +0100 |
commit | 1ddb2ed5fdc0a5e9944c5aeafcfc5ed4c07ea5cf (patch) | |
tree | a446c11f5fa8cf39f3037580c78fc5a32427e3e8 /ethosu/vela/npu_performance.py | |
parent | f8e353bd0a6e48f27e4c16b7243e403e5dae8d47 (diff) | |
download | ethos-u-vela-1ddb2ed5fdc0a5e9944c5aeafcfc5ed4c07ea5cf.tar.gz |
Vela: Fix diff in mean op
- Extend ifm/ofm dimensions explicitly in mean op
This fix a bug when ifm/ofm shape has different dimensions
e.g. IFM=1x19x18x25 axis=2 OFM=1x19x25,
the ofm_shape should be 1x19x1x25, not 1x1x19x25
- Fix wrong weight shape
Change-Id: I269eb71ea56c09deee2aa6c6433d9b2baa98a113
Signed-off-by: Diqing Zhong <diqing.zhong@arm.com>
Diffstat (limited to 'ethosu/vela/npu_performance.py')
0 files changed, 0 insertions, 0 deletions