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 ec766d5 add raise test for shape add d5666ed Round and sign straight-through-estimators C operators. (#16373) No new revisions were added by this update. Summary of changes: src/operator/contrib/stes_op.cc | 84 ++++++++++++++++ src/operator/contrib/stes_op.cu | 43 ++++++++ src/operator/contrib/stes_op.h | 33 +++++++ tests/python/unittest/test_contrib_stes_op.py | 137 ++++++++++++++++++++++++++ 4 files changed, 297 insertions(+) create mode 100644 src/operator/contrib/stes_op.cc create mode 100644 src/operator/contrib/stes_op.cu create mode 100644 src/operator/contrib/stes_op.h create mode 100644 tests/python/unittest/test_contrib_stes_op.py