torch-mlir/python/torch_mlir_e2e_test
Prashant Kumar 2b1b0f6e19 [LINALG] Add support for preserve memory format in aten_empty_like op.
The preserve memory specifies that `If any of the input tensors is in channels_last format,
operator output should be in channels_last format` and hence can be
added as is in aten_empty_like op.
2022-05-10 09:37:55 +05:30
..
linalg_on_tensors_backends llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
test_suite [LINALG] Add support for preserve memory format in aten_empty_like op. 2022-05-10 09:37:55 +05:30
torchscript Make e2e testing parallel 2022-05-05 21:27:58 -04:00
tosa_backends llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
CMakeLists.txt Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
__init__.py Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
utils.py Add a new `torch_mlir.compile` method. 2022-04-20 10:06:01 -07:00