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/Conversion/Utils
History
Vivek Khandelwal e90ea3d7ab [MLIR][TORCH] Extend implementation of aten._index_put_impl op. 
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>
 		2023-04-05 14:04:30 +05:30
..
CMakeLists.txt build: update llvm tag to 6f46ff37 (#1448) 2022-10-05 08:28:06 -05:00
Utils.cpp [MLIR][TORCH] Extend implementation of aten._index_put_impl op. 2023-04-05 14:04:30 +05:30