• 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
backend_test.in.cpp 264 KB