This is an automated email from the ASF dual-hosted git repository.
sxjscience pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git.
from bb6305d [MKLDNN] support mkldnn gelu (#16710)
add 5a2fce5 [WIP][New Op] Add deformable conv v2 (#16341)
No new revisions were added by this update.
Summary of changes:
python/mxnet/gluon/contrib/cnn/conv_layers.py | 180 ++++++-
...nl.h => modulated_deformable_convolution-inl.h} | 331 ++++++++-----
...tion.cc => modulated_deformable_convolution.cc} | 49 +-
...tion.cu => modulated_deformable_convolution.cu} | 19 +-
.../contrib/nn/modulated_deformable_im2col.cuh | 541 +++++++++++++++++++++
.../contrib/nn/modulated_deformable_im2col.h | 291 +++++++++++
tests/python/gpu/test_gluon_contrib_gpu.py | 27 +
tests/python/unittest/test_contrib_operator.py | 38 +-
tests/python/unittest/test_gluon_contrib.py | 30 ++
9 files changed, 1338 insertions(+), 168 deletions(-)
copy src/operator/contrib/{deformable_convolution-inl.h =>
modulated_deformable_convolution-inl.h} (54%)
copy src/operator/contrib/{deformable_convolution.cc =>
modulated_deformable_convolution.cc} (61%)
copy src/operator/contrib/{deformable_convolution.cu =>
modulated_deformable_convolution.cu} (68%)
create mode 100644 src/operator/contrib/nn/modulated_deformable_im2col.cuh
create mode 100644 src/operator/contrib/nn/modulated_deformable_im2col.h
