You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by bg...@apache.org on 2022/02/14 14:32:53 UTC
[incubator-mxnet] branch master updated (f4c4952 -> 4d915f7)
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 f4c4952 [FEATURE] Fuse dequantize with convolution (#20816)
add 4d915f7 Improve MaskedSoftmax by oneDNN (#20853)
No new revisions were added by this update.
Summary of changes:
src/operator/nn/dnnl/dnnl_base-inl.h | 4 +
src/operator/nn/dnnl/dnnl_masked_softmax-inl.h | 84 +++++++++++
src/operator/nn/dnnl/dnnl_masked_softmax.cc | 196 +++++++++++++++++++++++++
src/operator/nn/dnnl/dnnl_ops-inl.h | 6 +
src/operator/nn/dnnl/dnnl_softmax-inl.h | 40 -----
src/operator/nn/dnnl/dnnl_softmax.cc | 41 ++++++
src/operator/nn/masked_softmax.cc | 141 ++++++++++++++++++
src/operator/nn/softmax-inl.h | 16 ++
src/operator/nn/softmax.cc | 64 --------
9 files changed, 488 insertions(+), 104 deletions(-)
create mode 100644 src/operator/nn/dnnl/dnnl_masked_softmax-inl.h
create mode 100644 src/operator/nn/dnnl/dnnl_masked_softmax.cc
create mode 100644 src/operator/nn/masked_softmax.cc