mirror of https://github.com/llvm/torch-mlir
e90ea3d7ab
This commits adds the support for cases for index_put_op: 1.) where index is a 2-d tensor. 2.) where indices is a list of tensors and none, with exactly 2 non none tensors along the consecutive dimensions. This commit also adds a utility to compute the broadcast shape given the two input tensors. Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com> |
||
|---|---|---|
| .. | ||
| IR | ||
| Transforms | ||
| Utils | ||
| CMakeLists.txt | ||