Skip to content
Commit 1ddb2ed5 authored by Diqing Zhong's avatar Diqing Zhong
Browse files

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: default avatarDiqing Zhong <diqing.zhong@arm.com>
parent f8e353bd
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment