mirror of https://github.com/llvm/torch-mlir
400fecc1e5
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. |
||
|---|---|---|
| .. | ||
| CMakeLists.txt | ||
| DataMovement.cpp | ||
| IndirectDataMovement.cpp | ||
| Linear.cpp | ||
| Pooling.cpp | ||
| PopulatePatterns.h | ||
| Random.cpp | ||
| Reduction.cpp | ||
| TensorConstructors.cpp | ||
| TensorScalarInterop.cpp | ||
| TorchToLinalg.cpp | ||
| Uncategorized.cpp | ||
| Utils.cpp | ||
| Utils.h | ||