Skip to content
This repository has been archived by the owner on Apr 1, 2021. It is now read-only.

3D Convolution? #146

Open
xiuliren opened this issue Jan 2, 2020 · 1 comment
Open

3D Convolution? #146

xiuliren opened this issue Jan 2, 2020 · 1 comment

Comments

@xiuliren
Copy link

xiuliren commented Jan 2, 2020

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
Copy link

yinghai commented Jan 3, 2020

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

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants