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
test_layernorm2d_fp32.cpp 1.62 KB