llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/python/torch_mlir/dialects
History
Ramiro Leal-Cavazos 400fecc1e5
[LINALG] Fix shape function of index.Tensor + support N-rank inputs (#972) 
This commit fixes the shape function for `index.Tensor`, adding
support for multiple index tensors and `None`s in the indices
list. This commit also adds support for input tensors of rank greater
than 1. The lowering for `index.Tensor` still has the the limitation
that only a single index tensor along the first dimension of the input
tensor is supported.
 		2022-06-24 09:45:44 -07:00
..
torch [LINALG] Fix shape function of index.Tensor + support N-rank inputs (#972) 2022-06-24 09:45:44 -07:00
TorchBinding.td Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00