• Nick Korovaiko's avatar
    BatchNorm splitting into ops (2nd try) (#1828) · 1beec46b
    Nick Korovaiko authored
    * split bn into bn_inference bn_training
    
    * fix warnings
    
    * Add GPU support for the new BN ops (#1569)
    
    * Add GPU support and change batchnorm_globalstats test to use BNInference.
    
    * Changed test back to using BNTraining for global stats and updated cudnn backend to account for it.
    
    * Fix issues in merge with master.
    
    * Formatting.
    
    * CPU fixes
    
    * remove 5-arg training BN for now
    
    * more fixes
    
    * python batchnorm changes
    
    * fix onnx_import
    
    * fix a call BatchNormInference c-tor
    
    * yet another fix to BatchNormInference c-tor
    
    * AND yet another fix to batchnorm_inference c-tor
    
    * ops.py
    
    * address adam's feedback
    
    * Remove unnecessary parameter/argument.
    
    * remove batch_norm_training_relu_with_global_stats
    
    * remove bn_relu (training)
    1beec46b
Name
Last commit
Last update
..
examples Loading commit data...
ngraph Loading commit data...
pyngraph Loading commit data...
test Loading commit data...
.gitignore Loading commit data...
README.md Loading commit data...
requirements.txt Loading commit data...
setup.py Loading commit data...
test_requirements.txt Loading commit data...
tox.ini Loading commit data...