Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[Relay] Fix index order in conv2d computation for Arm CPU.
When dilation is larger than value 1 in conv2d with NHWC layout, the ordering of indexes when accessing data array in computation of convolution appears to be incorrect. 'data_vec' is defined as lambda n, oho, owo, kh, kw, ic, ohi, owi: But accessed as data_vec[n, oho, owo, kh, kw, ohi, owi, ic] This patch fixes the order of indexes.
- Loading branch information