You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@mxnet.apache.org by zh...@apache.org on 2021/04/14 00:27:10 UTC

[incubator-mxnet] branch master updated (e204dad -> f591b62)

This is an automated email from the ASF dual-hosted git repository.

zhasheng pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git.


    from e204dad  Define NVML_NO_UNVERSIONED_FUNC_DEFS (#20146)
     add f591b62  [FFI] part5: npx.batch_dot, npx.arange_like, npx.broadcast_like (#20110)

No new revisions were added by this update.

Summary of changes:
 python/mxnet/base.py                               |   3 +-
 python/mxnet/ndarray/numpy_extension/_op.py        | 132 +++++++++++++++++++-
 python/mxnet/numpy_extension/_op.py                | 134 ++++++++++++++++++++-
 .../{npx_one_hot_op.cc => npx_arange_like_op.cc}   |  56 +++++----
 .../{npx_pick_op.cc => npx_batch_dot_op.cc}        |  64 +++++-----
 ...ly_connected_op.cc => npx_broadcast_like_op.cc} |  47 ++++----
 src/operator/tensor/broadcast_reduce_op.h          |  15 +++
 src/operator/tensor/dot-inl.h                      |  28 +++++
 src/operator/tensor/init_op.h                      |  11 ++
 tests/python/unittest/test_numpy_op.py             |  10 ++
 10 files changed, 423 insertions(+), 77 deletions(-)
 copy src/api/operator/numpy_extension/{npx_one_hot_op.cc => npx_arange_like_op.cc} (63%)
 copy src/api/operator/numpy_extension/{npx_pick_op.cc => npx_batch_dot_op.cc} (62%)
 copy src/api/operator/numpy_extension/{npx_fully_connected_op.cc => npx_broadcast_like_op.cc} (66%)