• J
    oneDNN NHWC fixes (#40049) · dde9cec0
    Jacek Czaja 提交于
    * - Prototype of third solution
    
    - fix
    
    - compilation fixes
    
    - fix
    
    - fixe
    
    - fix
    
    - fix
    
    - compilation fix
    
    - comment fix
    
    - lint
    
    update mkldnn conv_elementwise_add_fuse_pass ut
    
    - NHWC changes to prelu
    
    - alhpa dims
    
    - UT fix
    
    - fix to UT
    
    - lint
    
    - Some fixes
    
    - added to BWD of prelu NHWC support
    
    - reverted removal of resetting cu_layout in clearing of caching
    
    * - Small changes
    
    * - compilation fix
    
    * - fix
    
    * - fix
    
    * lint
    
    * - fixes after internal review
    
    * - compilation fix
    
    * - lint
    dde9cec0
prelu_mkldnn_op.cc 7.8 KB