Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • rocking5566's avatar
    Refine layernorm naming and test code (#497) · d4d1147f
    rocking5566 authored 
    * Sync the naming

* Sync the test of layernorm with groupnorm

* Sync the naming

* Minor change for comment and log

* [What] Add saveMean and SaveInvVariance in the interface.
[Why] These can optimize the backward
    d4d1147f
layernorm2d.cpp 6.57 KB