mirror of https://github.com/llvm/torch-mlir
51d4d55f8a
This commit adds support for multi-dimensional tensors as input to the `_index_put_impl_` op. The support was to some degree already there, since `ScatterOp` already supports multi-dimensional tensors. This commit also adds a bit more error checking to `index_put` and refactors the code for creating `ScatterOp`s to mimic the way one would make a `Linalg::GenericOp`. |
||
|---|---|---|
| .. | ||
| AdjustCallingConventions.cpp | ||
| CMakeLists.txt | ||
| DecomposeComplexOps.cpp | ||
| DropShapeCalculations.cpp | ||
| GlobalizeObjectGraph.cpp | ||
| InlineGlobalSlots.cpp | ||
| MaximizeValueSemantics.cpp | ||
| PassDetail.h | ||
| Passes.cpp | ||
| PrepareForGlobalizeObjectGraph.cpp | ||
| ReduceOpVariants.cpp | ||
| RefinePublicReturn.cpp | ||
| RefineTypes.cpp | ||
| ReifyShapeCalculations.cpp | ||
| ShapeLibrary.cpp | ||
| SimplifyShapeCalculations.cpp | ||