torch-mlir

Commit Graph

Author	SHA1	Message	Date
Gaurav Shukla	3c40539b34	[TORCH][MLIR] Add E2E support for `aten.[ones_like\|zeros_like]` - This commit adds E2E support for `aten.ones_like` and `aten.zeros_like` ops. - Adds support for non-None `dtype` argument of `aten.empty_like` op. - All the unit test cases related to constant tensor allocation like ops are moved to a different file named `constant_alloc.py`. Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>	2022-01-06 20:24:40 +05:30

Author

SHA1

Message

Date

Gaurav Shukla

3c40539b34

[TORCH][MLIR] Add E2E support for `aten.[ones_like|zeros_like]`

- This commit adds E2E support for `aten.ones_like` and
  `aten.zeros_like` ops.
- Adds support for non-None `dtype` argument of `aten.empty_like` op.
- All the unit test cases related to constant tensor allocation like ops
  are moved to a different file named `constant_alloc.py`.

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>

2022-01-06 20:24:40 +05:30

1 Commits (c9a343267c863186d5127fb8e38a76f62a53e2ac)