• Amy Zhuang's avatar
    [MLIR] Use call back for MatMul. (#3838) · c737a573
    Amy Zhuang authored
    * [MLIR] Use call back for MatMul.
    
    * Use callback for Gemm.
    
    * Use mkldnn callback for Softmax.
    
    * Address PR feedback.
    
    * Fix merge errors.
    
    * Change to tail allocation struct.
    
    * Use mkldnn callback for AvgPool.
    
    * Add callbacks for AvgPoolBackprop, MaxPool, and MaxPoolBackprop.
    
    * Fix merge errors.
    
    * Use UnrankedMemRefType for callbacks.
    
    * Address PR feedback.
    
    * Cleanup.
    
    * Address PR feedback.
    
    * Fix a bug.
    
    * Use global variable to hold attributes.
    
    * Convert layout if needed for pooling.
    
    * Address PR feedback.
    
    * Add header.
    
    * Address PR feedback.
    
    * Update Copyright to 2017-2020.
    
    * Address PR feedback.
    Co-authored-by: 's avatarScott Cyphers <diyessi@users.noreply.github.com>
    c737a573
matmul.in.cpp 9.17 KB