• Pruthvi's avatar
    Fix bn constructor (#631) · 429eae9a
    Pruthvi authored
    * Fix bn construtor
        - assert if gamma or beta dont have rank 1
        - remove redundant checks
    
    * - added gaurds to check if the input and delta shape to mkldnn bn fprop and bprop op has a rank of 4
    429eae9a
Name
Last commit
Last update
cmake Loading commit data...
contrib/docker Loading commit data...
doc Loading commit data...
maint 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...
CMakeLists.txt Loading commit data...
INSTALL Loading commit data...
LICENSE Loading commit data...
README.md Loading commit data...
changes.md Loading commit data...