search

pytorch / tvm

TVM integration into PyTorch
453 stars 64 forks source link

3D Convolution? #146

Open xiuliren opened 4 years ago

xiuliren commented 4 years ago

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

yinghai commented 4 years ago

Yeah, we currently don't have plans for that. If you are interested, please add a patch. :)