torch-mlir/lib/Conversion/Utils
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