https://discuss.tvm.apache.org/t/check-failed-axes-defined-static-cast-int-axes-size-ndim-dimension-mismatch/9887 This seems a recurring issue. A Relay pass shouldn't change the input mod inplace.