-
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 |
|---|---|---|
| .ci | ||
| cmake | ||
| contrib/docker | ||
| doc | ||
| licenses | ||
| maint | ||
| python | ||
| src | ||
| test | ||
| .clang-format | ||
| .gitattributes | ||
| .gitignore | ||
| .gitmodules | ||
| .travis.yml | ||
| CMakeLists.txt | ||
| CODEOWNERS | ||
| CONTRIB.md | ||
| INSTALL.md | ||
| LICENSE | ||
| README.md | ||
| VERSION.in | ||
| changes.md |