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
