• Shucai Xiao's avatar
    Gpu batchnorm (#564) · 70ba8213
    Shucai Xiao authored
    
    
    * Initial cpu conv-nd
    
    * Formatting
    
    * Make index signed
    
    * Formatting
    
    * Assert the indices are greater than 0
    
    * Use equal instead of lexicographical_compare
    
    * Formatting
    
    * change the batchnorm cpu implementation to support multiple input dimensions
    
    * clang format
    
    * add unit tests for cpu batch_norm nd implementation
    
    * clang format
    
    * support nd batchnormalization
    
    * clang format
    
    * add rewrite batch_norm unit tests
    
    * clang format
    
    * remove a unit test
    
    * Fix tidy errors
    
    * Formatting
    
    * Handle different types
    
    * Formatting
    
    * Fix nested visits
    
    * Formatting
    
    * Add 3d conv test
    
    * Formatting
    
    * revert unnecessary changes
    
    * remove a print line
    
    * Fix ICE
    
    * Formatting
    
    * fix the per_activation mode of 2d
    
    * clang format
    
    * code clean up
    
    * clang format
    
    * add 1d and 3d gpu unit test
    
    * clang format
    
    * add unit test for rewrite_batchnorm
    
    * clang format
    
    * additional refinement
    
    * fix review comments
    
    * added a unit test to have more code coverage
    Co-authored-by: default avatarPaul <pfultz2@yahoo.com>
    Co-authored-by: default avatarmvermeulen <5479696+mvermeulen@users.noreply.github.com>
    70ba8213
rewrite_batchnorm.cpp 2.33 KB