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> |
||
|---|---|---|
| .. | ||
| CAPI | ||
| Conversion | ||
| Dialect | ||
| RefBackend | ||
| python | ||
| CMakeLists.txt | ||
| lit.cfg.py | ||
| lit.site.cfg.py.in | ||