My model is seq2seq. I also find the code for random_uniform is working fine, the error caused by loop control flow.
error info: shape mismatch ``` ...... %1817 = add(%35, %1816); %1818 = reshape(%1817, newshape(-1,[28500])) %1819 = topk(%1818, k=3, dtype='int32') %1820 = %1819.1 %1821 = cast(%1820, dtype='float64') %1822 = cast(9500, dtype='float64') %1823 = divide(%1821, %1822) %1824 = cast(%1823, dtype='int32') %1825 = add(%33, %1824) %1826 = reshape(%1825, newshape=[-1]) %1827 = take(%30, %1826, axis=0) in particular dimension 2 conflicts (int64)3 does not match 2; unable to unify: `Tensor[(15, 8, 3, 64), float32]` and `Tensor[(15, 8, 2, 64), float32]`; ; %1818 = take(%loop_var172, %loop_var50, axis =0) ........ ``` The %1827 node is 'while/GatherV2' the next node is NextIteration , so I suspect there's something wrong with the shape calculation in the loop. https://discuss.tvm.ai/t/relay-what-does-the-mistake-mean-in-particular-dimensio-2-conflicts-3-does-not-match-2/6149 --- [Visit Topic](https://discuss.tvm.ai/t/support-for-tensorflow-ops-for-newer-tf-version/2324/19) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.tvm.ai/email/unsubscribe/f529eb27105325a8f4be0e05b682c15f9b1542a2bcbb0acb22755362c632ffc1).
