llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/Dialect/Torch/Utils
History
Ramiro Leal-Cavazos 51d4d55f8a
Add support for multi-dim input to `index_put_impl` (#722) 
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`.
 		2022-03-31 09:27:21 -07:00
..
CMakeLists.txt Add type promotion code to refine types. 2021-10-29 11:17:39 -04:00
TorchUpstream.cpp Add type promotion code to refine types. 2021-10-29 11:17:39 -04:00
Utils.cpp Add support for multi-dim input to `index_put_impl` (#722) 2022-03-31 09:27:21 -07:00