Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

3D Deconvolution CPU support #19904

Closed
aulombard opened this issue Feb 16, 2021 · 2 comments · Fixed by #20292
Closed

3D Deconvolution CPU support #19904

aulombard opened this issue Feb 16, 2021 · 2 comments · Fixed by #20292

Comments

@aulombard
Copy link

Description

Until today mxnet does not support 3D Deconvolution (Transposed convolution) layers on CPU (even with mkldnn support) as it is only implemented for GPU with Cudnn support. Is there any plan in the future to support it knowing that mkldnn is supposed to support it and knowing that most of other frameworks support this feature. This layer is used a lot in UNet like networks, which is a very well known architecture used in computer vision problems.

References

@github-actions
Copy link

Welcome to Apache MXNet (incubating)! We are on a mission to democratize AI, and we are glad that you are contributing to it by opening this issue.
Please make sure to include all the relevant context, and one of the @apache/mxnet-committers will be here shortly.
If you are interested in contributing to our project, let us know! Also, be sure to check out our guide on contributing to MXNet and our development guides wiki.

@anko-intel
Copy link
Contributor

@mxnet-label-bot add [MKLDNN]

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