• A
    Add oneDNN fusion_gru kernel (#25594) · 68c6160e
    Adam 提交于
    * Add oneDNN fusion_gru kernel and fix fc+gru pass
    test=develop
    
    * Formatting changes
    test=develop
    
    * Lint fixes
    test=develop
    
    * Add memory::format_tag::any to GRU weights
    test=develop
    
    * Fix build with CUDA
    
    * Fix build with CUDA v2
    68c6160e
test_fusion_gru_op.py 3.9 KB