llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/python
History
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
..
test Fix kwarg annotation in eager (#747) 2022-04-11 17:35:42 -05:00
torch_mlir Update torch_mlir_tensor.py 2022-05-07 21:46:10 -04:00
torch_mlir_e2e_test [LINALG] Add support for preserve memory format in aten_empty_like op. 2022-05-10 09:37:55 +05:30
CMakeLists.txt Add a new `torch_mlir.compile` method. 2022-04-20 10:06:01 -07:00
TorchMLIRModule.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00