• Pruthvi's avatar
    Support for 5D batchnorm (#2055) · d4f8bfdc
    Pruthvi authored
    * - modified cpu_assignment pass to support bn with input 5D
    - added test cases for 5D bn and 5D bn+relu
    
    * - Address PR comments
    - used mkldnn_utils to validate bn for mkldnn
    
    * fix compilation error
    
    * Addressed PR comments
    - added helpers in mkldnn_utils for assigning ngraph Op as MKLDNN op
    - helper funnction for bn mkldnn assignment
    
    * fix clang error
    d4f8bfdc
Name
Last commit
Last update
.ci Loading commit data...
cmake Loading commit data...
contrib/docker Loading commit data...
doc Loading commit data...
licenses Loading commit data...
maint Loading commit data...
python Loading commit data...
src Loading commit data...
test Loading commit data...
.clang-format Loading commit data...
.gitattributes Loading commit data...
.gitignore Loading commit data...
.gitmodules Loading commit data...
.travis.yml Loading commit data...
ABOUT.md Loading commit data...
CMakeLists.txt Loading commit data...
CODEOWNERS Loading commit data...
CONTRIB.md Loading commit data...
FAQs.md Loading commit data...
INSTALL.md Loading commit data...
LICENSE Loading commit data...
README.md Loading commit data...
VERSION.in Loading commit data...
changes.md Loading commit data...
ecosystem-overview.md Loading commit data...