This is an automated email from the ASF dual-hosted git repository.
haoj pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git.
from 2ad3ce4 broadcast_axis optimization (#17091)
add 37197b6 any/all (#17087)
No new revisions were added by this update.
Summary of changes:
python/mxnet/_numpy_op_doc.py | 101 +++++++++++++++++++++
python/mxnet/numpy_dispatch_protocol.py | 2 +
src/operator/mshadow_op.h | 8 ++
src/operator/mxnet_op.h | 23 +++++
src/operator/numpy/np_broadcast_reduce_op.h | 53 +++++++++++
.../numpy/np_broadcast_reduce_op_boolean.cc | 85 +++++++++++++++++
..._index.cu => np_broadcast_reduce_op_boolean.cu} | 15 +--
src/operator/operator_tune.cc | 2 +
src/operator/tensor/broadcast_reduce-inl.cuh | 10 ++
src/operator/tensor/broadcast_reduce-inl.h | 12 +++
src/operator/tensor/broadcast_reduce_op.h | 33 +++++++
src/operator/tensor/matrix_op-inl.h | 2 +-
.../python/unittest/test_numpy_interoperability.py | 28 ++++++
tests/python/unittest/test_numpy_op.py | 78 ++++++++++++++++
14 files changed, 445 insertions(+), 7 deletions(-)
create mode 100644 src/operator/numpy/np_broadcast_reduce_op_boolean.cc
copy src/operator/numpy/{np_broadcast_reduce_op_index.cu => np_broadcast_reduce_op_boolean.cu}
(75%)
|