torch-mlir/lib/Conversion/TorchToLinalg
Vivek Khandelwal 6844c84702
[MLIR][Torch] Fix OnnxToLinalg lowering for AvgPool op (#3076)
Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
2024-04-01 22:14:14 +05:30
..
CMakeLists.txt Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
DataMovement.cpp Implement linalg lowering of diag_embed torch op (#2885) 2024-03-22 16:32:50 -07:00
IndirectDataMovement.cpp Bump LLVM to llvm/llvm-project@bb180856ec (#2895) 2024-02-09 14:07:49 -08:00
Linear.cpp Adds accumulator types in TorchToLinalg for `AtenMmOp` and `AtenConvolutionOp` (#3027) 2024-03-14 16:40:40 -07:00
Pooling.cpp [MLIR][Torch] Fix OnnxToLinalg lowering for AvgPool op (#3076) 2024-04-01 22:14:14 +05:30
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 [onnx] Migrate `onnx.ReduceMax` to match `onnx.ReduceMin` (#2981) 2024-03-06 16:48:21 -08:00
TensorConstructors.cpp [onnx] Fix constant pad for dynamic shape (#2989) 2024-03-07 13:29:50 -08:00
TensorScalarInterop.cpp [torch] Improve shape inference for `torch-to-linalg` path for reshapes (#3055) 2024-03-26 12:41:40 -07:00
TorchToLinalg.cpp Add complex types support with basic complex ops. 2023-05-11 21:29:07 +05:30
Uncategorized.cpp [torch] Fix clamp ranges on quantize_per_tensor on unsigned (#3018) 2024-03-20 13:37:47 -07:00
Utils.cpp Bump LLVM to llvm/llvm-project@bb180856ec (#2895) 2024-02-09 14:07:49 -08:00