torch-mlir/lib/Conversion/TorchToLinalg
Prashant Kumar b1a506624c Add decomposition of `aten.masked.tensor` op.
`aten.masked.tensor` op has been decomposed to `aten.masked.scalar` op.
2022-08-11 07:48:04 +05:30
..
CMakeLists.txt Bump LLVM (#958) 2022-06-22 22:23:46 -05:00
DataMovement.cpp llvm: update tag to 061e0189 (#1180) 2022-08-08 20:17:35 -07:00
IndirectDataMovement.cpp E2E support for AtenEmbeddingBagPaddingIdxOp SUM Mode (#1066) 2022-08-01 16:44:11 -04:00
Linear.cpp Add non-unit groups support to aten.convolution (#858) 2022-08-04 02:18:38 -04:00
Pooling.cpp [MLIR][TORCH] Add decomposition of aten.adaptive_avg_pool2d op 2022-05-27 07:56:37 +05:30
PopulatePatterns.h Temporarily revert support for custom op extensions. (#944) 2022-06-14 18:24:40 -07:00
Random.cpp Improve "pseudo" op terminology. 2022-03-15 17:57:52 -07:00
Reduction.cpp [MLIR][TORCH} Fix empty dim cases for the .dim ops 2022-07-29 11:08:57 +05:30
TensorConstructors.cpp [LINALG] Add support for preserve memory format in aten_empty_like op. 2022-05-10 09:37:55 +05:30
TensorScalarInterop.cpp [MLIR][TORCH] Add E2E support for ScalarImplicit, Int.Scalar op 2022-05-10 22:40:49 +05:30
TorchToLinalg.cpp Temporarily revert support for custom op extensions. (#944) 2022-06-14 18:24:40 -07:00
Uncategorized.cpp Add decomposition of `aten.masked.tensor` op. 2022-08-11 07:48:04 +05:30
Utils.cpp [LINALG] Lower `aten.Matmul` to `linalg.BatchMatmul` 2022-06-25 10:58:06 +05:30
Utils.h [LINALG] Lower `aten.Matmul` to `linalg.BatchMatmul` 2022-06-25 10:58:06 +05:30