This is an automated email from the ASF dual-hosted git repository.
apeforest pushed a change to branch benchmark
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git.
from 8c22fac Aggregated adamw update (#16398)
add 82ed82f Aggregated zero grad (#16446)
No new revisions were added by this update.
Summary of changes:
python/mxnet/gluon/parameter.py | 21 ++++-
src/operator/contrib/reset_arrays-inl.h | 92 ++++++++++++++++++++++
src/operator/contrib/reset_arrays.cc | 74 +++++++++++++++++
.../reset_arrays.cu} | 16 ++--
tests/python/unittest/test_gluon.py | 66 +++++++++++++---
5 files changed, 252 insertions(+), 17 deletions(-)
create mode 100644 src/operator/contrib/reset_arrays-inl.h
create mode 100644 src/operator/contrib/reset_arrays.cc
copy src/operator/{numpy/np_broadcast_reduce_op_index.cu =>
contrib/reset_arrays.cu} (69%)
