• 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
tf_conv_mnist_nhwc.json 8.34 KB