[GitHub] [incubator-mxnet] wkcn commented on a change in pull request #18504: [Improvement] Invoke mkldnn and cudnn BatchNorm when axis != 1

wkcn commented on a change in pull request #18504: URL: https://github.com/apache/incubator-mxnet/pull/18504#discussion_r449862796 ## File path: src/operator/nn/cudnn/cudnn_batch_norm-inl.h ## @@ -262,15 +262,26 @@ class CuDNNBatchNormOp { private: void Init(const TBlob

[GitHub] [incubator-mxnet] wkcn commented on a change in pull request #18504: [Improvement] Invoke mkldnn and cudnn BatchNorm when axis != 1

wkcn commented on a change in pull request #18504: URL: https://github.com/apache/incubator-mxnet/pull/18504#discussion_r449849259 ## File path: src/operator/nn/cudnn/cudnn_batch_norm-inl.h ## @@ -262,15 +262,26 @@ class CuDNNBatchNormOp { private: void Init(const TBlob

[GitHub] [incubator-mxnet] wkcn commented on a change in pull request #18504: [Improvement] Invoke mkldnn and cudnn BatchNorm when axis != 1

wkcn commented on a change in pull request #18504: URL: https://github.com/apache/incubator-mxnet/pull/18504#discussion_r449349061 ## File path: src/operator/nn/cudnn/cudnn_batch_norm-inl.h ## @@ -262,15 +262,26 @@ class CuDNNBatchNormOp { private: void Init(const TBlob

[GitHub] [incubator-mxnet] wkcn commented on a change in pull request #18504: [Improvement] Invoke mkldnn and cudnn BatchNorm when axis != 1

wkcn commented on a change in pull request #18504: URL: https://github.com/apache/incubator-mxnet/pull/18504#discussion_r449349061 ## File path: src/operator/nn/cudnn/cudnn_batch_norm-inl.h ## @@ -262,15 +262,26 @@ class CuDNNBatchNormOp { private: void Init(const TBlob

[GitHub] [incubator-mxnet] wkcn commented on a change in pull request #18504: [Improvement] Invoke mkldnn and cudnn BatchNorm when axis != 1

wkcn commented on a change in pull request #18504: URL: https://github.com/apache/incubator-mxnet/pull/18504#discussion_r449349061 ## File path: src/operator/nn/cudnn/cudnn_batch_norm-inl.h ## @@ -262,15 +262,26 @@ class CuDNNBatchNormOp { private: void Init(const TBlob

[GitHub] [incubator-mxnet] wkcn commented on a change in pull request #18504: [Improvement] Invoke mkldnn and cudnn BatchNorm when axis != 1

wkcn commented on a change in pull request #18504: URL: https://github.com/apache/incubator-mxnet/pull/18504#discussion_r449341276 ## File path: src/operator/nn/batch_norm.cc ## @@ -420,10 +420,14 @@ static bool BatchNormType(const nnvm::NodeAttrs& attrs, #if MXNET_USE_MKLDN

[GitHub] [incubator-mxnet] wkcn commented on a change in pull request #18504: [Improvement] Invoke mkldnn and cudnn BatchNorm when axis != 1

wkcn commented on a change in pull request #18504: URL: https://github.com/apache/incubator-mxnet/pull/18504#discussion_r449329691 ## File path: src/operator/nn/batch_norm.cc ## @@ -420,10 +420,14 @@ static bool BatchNormType(const nnvm::NodeAttrs& attrs, #if MXNET_USE_MKLDN

[GitHub] [incubator-mxnet] wkcn commented on a change in pull request #18504: [Improvement] Invoke mkldnn and cudnn BatchNorm when axis != 1

wkcn commented on a change in pull request #18504: URL: https://github.com/apache/incubator-mxnet/pull/18504#discussion_r449329691 ## File path: src/operator/nn/batch_norm.cc ## @@ -420,10 +420,14 @@ static bool BatchNormType(const nnvm::NodeAttrs& attrs, #if MXNET_USE_MKLDN

[GitHub] [incubator-mxnet] wkcn commented on a change in pull request #18504: [Improvement] Invoke mkldnn and cudnn BatchNorm when axis != 1

wkcn commented on a change in pull request #18504: URL: https://github.com/apache/incubator-mxnet/pull/18504#discussion_r449326728 ## File path: src/operator/nn/cudnn/cudnn_batch_norm-inl.h ## @@ -262,15 +262,26 @@ class CuDNNBatchNormOp { private: void Init(const TBlob