wkcn commented on pull request #18500: URL: https://github.com/apache/incubator-mxnet/pull/18500#issuecomment-640178213
Hi @sxjscience , @zhreshold and @TaoLv , could you please help take a review? The bug of batch norm has been fixed. I change the following implementations of batch norm. name | filename -|- naive cpu | src/operator/nn/batch_norm.cc naive gpu | src/operator/nn/batch_norm.cu cudnn|src/operator/nn/cudnn/cudnn_batch_norm-inl.h mkldnn | src/operator/nn/mkldnn/mkldnn_batch_norm-inl.h Thank you! ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected]
