rongzha1 commented on a change in pull request #17170: add mkldnn softmax
backward
URL: https://github.com/apache/incubator-mxnet/pull/17170#discussion_r376709516
##
File path: src/operator/nn/mkldnn/mkldnn_softmax.cc
##
@@ -131,6 +143,80 @@ void MKLDNNSoftmaxForward(cons
rongzha1 commented on a change in pull request #17170: add mkldnn softmax
backward
URL: https://github.com/apache/incubator-mxnet/pull/17170#discussion_r368829691
##
File path: src/operator/nn/mkldnn/mkldnn_softmax.cc
##
@@ -131,6 +143,80 @@ void MKLDNNSoftmaxForward(cons
rongzha1 commented on a change in pull request #17170: add mkldnn softmax
backward
URL: https://github.com/apache/incubator-mxnet/pull/17170#discussion_r366658974
##
File path: src/operator/nn/mkldnn/mkldnn_softmax.cc
##
@@ -131,6 +143,80 @@ void MKLDNNSoftmaxForward(cons
rongzha1 commented on a change in pull request #17170: add mkldnn softmax
backward
URL: https://github.com/apache/incubator-mxnet/pull/17170#discussion_r366658426
##
File path: src/operator/nn/mkldnn/mkldnn_softmax.cc
##
@@ -131,6 +143,80 @@ void MKLDNNSoftmaxForward(cons
rongzha1 commented on a change in pull request #17170: add mkldnn softmax
backward
URL: https://github.com/apache/incubator-mxnet/pull/17170#discussion_r361586715
##
File path: src/operator/nn/mkldnn/mkldnn_softmax.cc
##
@@ -86,6 +98,33 @@ void MKLDNNSoftmaxForward(const
rongzha1 commented on a change in pull request #17170: add mkldnn softmax
backward
URL: https://github.com/apache/incubator-mxnet/pull/17170#discussion_r361585996
##
File path: src/operator/nn/mkldnn/mkldnn_softmax.cc
##
@@ -86,6 +98,33 @@ void MKLDNNSoftmaxForward(const
rongzha1 commented on a change in pull request #17170: add mkldnn softmax
backward
URL: https://github.com/apache/incubator-mxnet/pull/17170#discussion_r361566549
##
File path: src/operator/nn/softmax.cc
##
@@ -54,13 +54,30 @@ static void SoftmaxComputeExCPU(const nnvm::N
rongzha1 commented on a change in pull request #17170: add mkldnn softmax
backward
URL: https://github.com/apache/incubator-mxnet/pull/17170#discussion_r361566529
##
File path: src/operator/nn/mkldnn/mkldnn_softmax.cc
##
@@ -86,6 +98,33 @@ void MKLDNNSoftmaxForward(const
rongzha1 commented on a change in pull request #17170: add mkldnn softmax
backward
URL: https://github.com/apache/incubator-mxnet/pull/17170#discussion_r361566500
##
File path: src/operator/nn/mkldnn/mkldnn_ops-inl.h
##
@@ -93,6 +93,10 @@ void MKLDNNLeakyReluBackward(cons
rongzha1 commented on a change in pull request #17170: add mkldnn softmax
backward
URL: https://github.com/apache/incubator-mxnet/pull/17170#discussion_r361566556
##
File path: src/operator/nn/softmax.cc
##
@@ -54,13 +54,30 @@ static void SoftmaxComputeExCPU(const nnvm::N
rongzha1 commented on a change in pull request #17170: add mkldnn softmax
backward
URL: https://github.com/apache/incubator-mxnet/pull/17170#discussion_r361566537
##
File path: src/operator/nn/mkldnn/mkldnn_softmax.cc
##
@@ -86,6 +98,33 @@ void MKLDNNSoftmaxForward(const
11 matches
Mail list logo