3D Convolution?
xiuliren opened this issue · comments
It seems that there is no 3D convolution support. the tests are 1D and 2D. Is it correct? if so, are there plan to add 3D support? 3D convolution is popular in medical image processing now.
TVM has support for 3D convolution:
https://github.com/apache/incubator-tvm/blob/1ef1605a378c5a44f645b2f1459a58e0092f8726/topi/tests/python/test_topi_conv3d_ncdhw.py
Yeah, we currently don't have plans for that. If you are interested, please add a patch. :)