[TVM Discuss] [Development] Loop partitioning, padding and tensorization

2020-08-28 Thread Giuseppe Rossini via TVM Discuss
Hi Animesh, The problem is that I need padding added in the middle of TIR on my (transformed) data tensor. I.e., something like ``` A1 = im2col(A) A2 = pad(A1) C_padded = te.compute([M,N], lambda i, j : sum(A2[i,k]*B[k,j], k) C = unpad(C)+requantization ``` Then I tile on `C` and tensorize o

[TVM Discuss] [Development] Loop partitioning, padding and tensorization

2020-08-28 Thread Animesh Jain via TVM Discuss
How about using Relay Legalize pass to add an explicit padding at the graph level? --- [Visit Topic](https://discuss.tvm.ai/t/loop-partitioning-padding-and-tensorization/7753/2) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [

[TVM Discuss] [Development] Loop partitioning, padding and tensorization

2020-08-28 Thread Giuseppe Rossini via TVM Discuss
Hi all, In my effort to accelerate AArch64 through tensorization, I incurred into an issue. Basically, I am padding my input tensor, to let `tensorize` work (I need rows to be multiple of 4 and cols to be multiple of 16). However, bound inference removes padding (since it is not used) and