torch-mlir/python/torch_mlir/dialects/torch
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
..
importer Add shape info to `rand_like` + support for `dtype` flag (#851) 2022-05-12 16:00:59 -07:00
__init__.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00