• 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
..
ngraph Loading commit data...
resource Loading commit data...
tools Loading commit data...
CMakeLists.txt Loading commit data...