[BACKPORT][BUGFIX][FEATURE] Add oneDNN 1D and 3D deconvolution support and fix bias (#20292)
* [v1.x][BUGFIX] Implement oneDNN deconvolution primitives to deconvolution 2D (#20107) * Use mkldnn deconvolution primitive in deconvolution * Apply clang-format * Refactor deconvolution version 1 * Refactor deconvolution version 2 and use permute_axes in IOLogicalSwapDesc * Refactor deconvolution version 3 * Enable Deconvolution2D test * Fix sanity * Fix windows builds * Fix deconvolution with bias test * [v1.x][FEATURE] Add MKLDNN Deconvolution 1D and 3D support (#20137) * Use MXNET_USE_ONEDNN * Fix test * Apply formatter * Add native support for 3D deconvolution * Remove outdated check * Replace math.prod with np.prod * Check convolution layout only when it has value * Remove outdated check * Change tests * Increase default workspace size to mach convolution * Fix deconv workspace size * Increase default deconv workspace size in python API * Disable 3D tests for GPU * Add deconv arguments checks * Remove next_impl calls until it is fixed * Share workspace * Fix documentation * Add test_deconv_dilation * Fix check * Fix include order
P
Paweł Głomski committed
ab222114afb234bce6ebffb99b09e2b16d2ddcec
Parent: 80d72b5
Committed by GitHub <noreply@github.com>
on 9/29/2021, 3:41:36 PM