tvm icon indicating copy to clipboard operation
tvm copied to clipboard
verified publisher icon pytorch

3D Convolution?

Open xiuliren opened this issue 5 years ago • 1 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

xiuliren avatar Jan 02 '20 15:01 xiuliren

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

yinghai avatar Jan 03 '20 16:01 yinghai