torch-mlir/lib/Conversion/TorchToLinalg
Rob Suderman 34f6948533
[torch] Support `!countIncludePad` when unpadded for average pool (#2836)
We do not support average pool when `countIncludePad is set to false.
However if the input is unpadded then the setting of the boolean is
unneeded. Extended use by checking if padding is zero before rejecting
the lowering.
2024-01-31 15:09:36 -08: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 Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
IndirectDataMovement.cpp Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
Linear.cpp [torch] Support `torch.convolution` quantized lowering to `linalg` (#2811) 2024-01-30 13:46:47 -08:00
Pooling.cpp [torch] Support `!countIncludePad` when unpadded for average pool (#2836) 2024-01-31 15:09:36 -08:00
PopulatePatterns.h Re-enable custom op support 2022-08-16 22:49:08 +05:30
Random.cpp Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
Reduction.cpp Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
TensorConstructors.cpp Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
TensorScalarInterop.cpp Elide dynamic broadcast checks when in strict symbolic shapes mode. (#2496) 2023-09-29 16:45:48 -07:00
TorchToLinalg.cpp Add complex types support with basic complex ops. 2023-05-11 21:29:07 +05:30
Uncategorized.cpp [torch] Support `torch.convolution` quantized lowering to `linalg` (#2811) 2024-01-30 13:46:47 -08:00
Utils.cpp [torch] Support `torch.convolution` quantized lowering to `linalg` (#2811) 2024-01-30 13:46:47 -08:00