• Amy Zhuang's avatar
    [MLIR] Use mkldnn callback for ConvBias. (#4205) · a8a9bcb5
    Amy Zhuang authored
    * [MLIR] Use mkldnn callback for ConvBias.
    
    * Add try catch.
    
    Fix opAttrsVec.
    
    Add rank check for Gemm and MatMul.
    
    * Fix merge error.
    
    * Fix a bug.
    
    * Fix lit test.
    
    * Modify unit test.
    
    * Fix merge error.
    
    * Address PR feedback.
    
    * Address PR feedback.
    
    * Insert callback_init function to module.
    
    * Fix lit tests.
    
    * Fix a bug.
    
    * Use a set of GlobalOps for attributes.
    
    * Address PR feedback.
    
    * Address PR feedback.
    
    * Fix merge error.
    
    * Fix style error.
    
    * Fix style error.
    Co-authored-by: 's avatarScott Cyphers <diyessi@users.noreply.github.com>
    a8a9bcb5
callback_ops.mlir 8.51 KB