torch-mlir/lib/Conversion/TorchToLinalg
zjgarvey f1c74e1431
[TorchToLinalg] add support for depthwise qconv (#3564)
- 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.
2024-07-29 12:25:07 -07:00
..
CMakeLists.txt Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
DataMovement.cpp Add support for multiple dynamic reassociation dims for unflatten.int (#3504) 2024-06-28 09:59:51 -07:00
IndirectDataMovement.cpp [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07:00
Linear.cpp [TorchToLinalg] add support for depthwise qconv (#3564) 2024-07-29 12:25:07 -07:00
Pooling.cpp [ONNX] Add averagepool dilations support (#3490) 2024-06-21 17:24:57 -07:00
PopulatePatterns.h Re-enable custom op support 2022-08-16 22:49:08 +05:30
Random.cpp [E2E][ONNX] torch.multinomial (#3404) 2024-07-16 23:09:39 +05:30
Reduction.cpp [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07:00
TensorConstructors.cpp Adds misc fixes for some padding related issues (#3528) 2024-07-11 20:01:45 -05:00
TensorScalarInterop.cpp [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07:00
TorchToLinalg.cpp [TorchToLinalg][ONNX] Add Basic Determinant Support (#3481) 2024-06-25 13:34:19 -05:00
Uncategorized.cpp Align Quantization Rounding Scheme with ONNX/Pytorch (#3569) 2024-07-29 12:24:46 -07:00
Utils.cpp Fix unused variable warning from assertion variable (#3512) 2024-06-28 12:20:29 -04:00