[MXNet] compiled mx.nd.take output is wrong

apivovarov · April 10, 2020, 2:44am

I prepared MXNet Hybrid block which uses F.take(x, [1,2], axis=1).

MXNet and TVM outputs are different.

Script and output are below:

gist.github.com

https://gist.github.com/apivovarov/7e951a86e9bda9d1b450b066839adea8

test_mxnet_take_op.py

#!/usr/bin/env python3
import numpy as np
import tvm
import mxnet as mx
from mxnet import nd
from mxnet.gluon import nn
from tvm import relay
import tvm.contrib.graph_runtime as runtime

dshape = (1,3,8,8)

This file has been truncated. show original

http://beta.mxnet.io/r/api/mx.nd.take.html

apivovarov · April 10, 2020, 7:40pm

@haichen What you think?

haichen · April 10, 2020, 8:21pm

I’ll take a look when I get some time.