"vscode:/vscode.git/clone" did not exist on "aca5129458f0cced4e32875a9742bfbbccc202ec"
  • 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