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
Ramiro Leal-Cavazos 96f90efd16
Add shape info to `rand_like` + support for `dtype` flag (#851) 
The op `aten.rand_like` was missing a shape function, unit tests, and
the `dtype` argument was being ignored in its decomposition. This
commit fixes all three things.
 		2022-05-12 16:00:59 -07:00
..
test Fix kwarg annotation in eager (#747) 2022-04-11 17:35:42 -05:00
torch_mlir Add shape info to `rand_like` + support for `dtype` flag (#851) 2022-05-12 16:00:59 -07:00
torch_mlir_e2e_test Add shape info to `rand_like` + support for `dtype` flag (#851) 2022-05-12 16:00:59 -07:00
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