access2rohit commented on issue #17805: fixing batch_norm and layer_norm for large tensors URL: https://github.com/apache/incubator-mxnet/pull/17805#issuecomment-597252004 @apeforest @ChaiBapchya I don't know much about layer_norm() or batch_norm() to add suitable shape checks in the tests. But is have provided gdb outputs after fixing the code. Can you guys suggest addition of proper shape testing that can be added at python end ?
---------------------------------------------------------------- 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] With regards, Apache Git Services
