torch-mlir

Commit Graph

Author	SHA1	Message	Date
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
Ramana Radhakrishnan	738f4fe96a	Rename TorchToStd pass as TorchToArith (#1163 ) All the converters in this pass appear to create ops from the arith dialect. Hence the full rename. Fix GH Issue #409.	2022-08-10 20:12:51 +01:00

Author

SHA1

Message

Date

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

Ramana Radhakrishnan

738f4fe96a

Rename TorchToStd pass as TorchToArith (#1163 )

All the converters in this pass appear to create ops from the
arith dialect. Hence the full rename.

Fix GH Issue #409.

2022-08-10 20:12:51 +01:00

2 Commits (wjw.emit_nonzero)