This is an automated email from the ASF dual-hosted git repository.
bgawrych pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git.
from 26f9fa6 Unifying oneDNN post-quantization properties (#20724)
add f67b222 Add oneDNN support for reduce operators (#20669)
No new revisions were added by this update.
Summary of changes:
src/operator/nn/dnnl/dnnl_reduce-inl.h | 108 ++++++++++
src/operator/nn/dnnl/dnnl_reduce.cc | 236 +++++++++++++++++++++
src/operator/nn/dnnl/dnnl_transpose-inl.h | 4 +-
src/operator/nn/dnnl/dnnl_transpose.cc | 5 +-
src/operator/numpy/np_broadcast_reduce_op.h | 19 ++
src/operator/numpy/np_broadcast_reduce_op_value.h | 53 +++++
.../numpy/np_broadcast_reduce_op_value_mean.cc | 5 +
.../numpy/np_broadcast_reduce_op_value_sum.cc | 5 +
src/operator/tensor/broadcast_reduce_op.h | 21 ++
src/operator/tensor/broadcast_reduce_sum_value.cc | 7 +
10 files changed, 459 insertions(+), 4 deletions(-)
create mode 100644 src/operator/nn/dnnl/dnnl_reduce-inl.h
create mode 100644 src/operator/nn/dnnl/dnnl_reduce.cc