mirror of https://github.com/llvm/torch-mlir
f1c74e1431
- Adds support for lowering depthwise + quantized convolution ops to linalg::DepthwiseConv2DNhwcHwcQOp - Changed the variable name for groupSize (which is really C/G) to the more appropriate numGroups (G). - Discovered in e2e testing that linalg does not accept (Cin = groups && Cout = K*groups for K>1) as a "depthwise" conv, so this also updates the case-checking to reflect this issue. |
||
|---|---|---|
| .. | ||
| test | ||
| torch_mlir | ||
| torch_mlir_e2e_test | ||
| CMakeLists.txt | ||