torch-mlir/python/torch_mlir
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
..
dialects Add shape info to `rand_like` + support for `dtype` flag (#851) 2022-05-12 16:00:59 -07:00
eager_mode Update torch_mlir_tensor.py 2022-05-07 21:46:10 -04:00
__init__.py Add tracing suport to `torch_mlir.compile`. 2022-05-03 09:08:40 -07:00
compiler_utils.py Add a new `torch_mlir.compile` method. 2022-04-20 10:06:01 -07:00