torch-mlir/lib/Conversion/TorchToLinalg
Prateek Gupta 2d75654b2c [TORCH][MLIR] Add lowering of `aten.slice_scatter` and
`aten.select_scatter` op.

This commit adds:
1.  Lowering of `aten.slice_scatter` op into `tensor.insert_slice`
op.
2. Decomposes the `aten.select_scatter` op into `aten.slice_scater`
op.

Signed-Off-By: Prateek Gupta <gprateek93@gmail.com>
2022-07-11 14:07:21 +05:30
..
CMakeLists.txt Bump LLVM (#958) 2022-06-22 22:23:46 -05:00
DataMovement.cpp [TORCH][MLIR] Add lowering of `aten.slice_scatter` and 2022-07-11 14:07:21 +05:30
IndirectDataMovement.cpp [LINALG] Fix shape function of index.Tensor + support N-rank inputs (#972) 2022-06-24 09:45:44 -07:00
Linear.cpp [LINALG] Lower `aten.Matmul` to `linalg.BatchMatmul` 2022-06-25 10:58:06 +05:30
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 torch,linalg: add support for translating aten.linalg.vector_norm (#839) 2022-05-19 15:48:15 -07:00
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 linalg: lower `aten.triu` op to `linalg.generic` (#965) 2022-06-23 22:45:48 -07:00
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