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/torch_mlir
History
Sean Silva ef9e4c95f2 torch_mlir.compile: add support for dynamic sizes. 
We do this by inroducing a TensorPlaceholder class, which can be used to
specify dynamic sizes. Internally, we canonicalize all example inputs
to TensorPlaceholder's.

This commit also adds some basic testing, which was missing before.
 		2022-05-17 07:02:32 -07:00
..
dialects mlir: bump llvm tag to 5380e3 (#856) 2022-05-16 12:54:35 -07:00
eager_mode Update torch_mlir_tensor.py 2022-05-07 21:46:10 -04:00
__init__.py torch_mlir.compile: add support for dynamic sizes. 2022-05-17 07:02:32 -07:00
compiler_utils.py Add a new `torch_mlir.compile` method. 2022-04-20 10:06:01 -07:00