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

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

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

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

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

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

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

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

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

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

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

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