1. 16 Jan, 2020 1 commit
    • 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