• Nagy Mostafa's avatar
    [MLIR] Enable lowering of GroupConv in MLIR CPU backend (#4102) · 8e46ff86
    Nagy Mostafa authored
    * WIP
    
    * WIP
    
    * Refactored existing convolution
    
    * Add Channel and num of filters bounds parameters to helper
    
    * Works on unit-tests. v1 op gets converted and breaks
    
    * Fixed group conv with groups in filters shape. Tests pass
    
    * style
    
    * add LIT tests
    
    * Switch outer loop to affine loop
    
    * re-org code
    
    * PR fixes
    
    * Revert ops.td
    
    * PR fixes
    8e46ff86
ops_supported.inc 628 Bytes