kpuatamazon opened a new pull request #19602: URL: https://github.com/apache/incubator-mxnet/pull/19602
## Description ## This is the master version of #19601. There isn't much different in LayerNormalization implementation between v1.x and master. ## Checklist ## ### Essentials ### - [x] PR's title starts with a category (e.g. [BUGFIX], [MODEL], [TUTORIAL], [FEATURE], [DOC], etc) - [x] Changes are complete (i.e. I finished coding on this PR) - [x] All changes have test coverage; already covered by unittest for layer normalization. - [x] Code is well-documented ### Changes ### - [x] Copy Marian optimized CPU LayerNorm implementation and adapt to MXNet. - [x] Refactor dispatch of optimized versions using bool return value. - [x] Remove MKL call ## Comments ## See #19601 for benchmarks. ---------------------------------------------------------------- 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]
