• Pruthvi's avatar
    Pruthvi/bn to support globalstats (#783) · 1d80cabe
    Pruthvi authored
    * WIP support bn training for global_stats
    
    (cherry picked from commit eb81a37328ea177b1d58c9eebdbb345e0fa25f0d)
    
    * - Style fix
    - Fix test case
    
    * Addressed PR comments
    - added support for bn training/inference with a same ctor
    - added more verbose comments in bn header
    
    * Fixed bn serializer and default value in bn ctor for bwd compatibility
    
    * proposed docs change
    
    * - Addressed PR comments
      - added support to compute bn inference/training using same mkldnn kernel with global stats
    
    * fix unit bn relu unit test
    1d80cabe
Name
Last commit
Last update
cmake Loading commit data...
contrib/docker Loading commit data...
doc Loading commit data...
licenses Loading commit data...
maint Loading commit data...
python Loading commit data...
src Loading commit data...
test Loading commit data...
third-party Loading commit data...
.clang-format Loading commit data...
.gitignore Loading commit data...
.gitmodules Loading commit data...
CMakeLists.txt Loading commit data...
INSTALL.md Loading commit data...
LICENSE Loading commit data...
README.md Loading commit data...
changes.md Loading commit data...