"vscode:/vscode.git/clone" did not exist on "980fb59d4b55813faeb4cd93e050f64a93a75e48"
  • 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
layernorm_blockwise.cpp 6.17 KB