tree a446c11f5fa8cf39f3037580c78fc5a32427e3e8
parent f8e353bd0a6e48f27e4c16b7243e403e5dae8d47
author Diqing Zhong <diqing.zhong@arm.com> 1646825027 +0100
committer Diqing Zhong <diqing.zhong@arm.com> 1647014895 +0100

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>