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/test
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
..
compile_api torch_mlir.compile: add support for dynamic sizes. 2022-05-17 07:02:32 -07:00
eager_mode Fix kwarg annotation in eager (#747) 2022-04-11 17:35:42 -05:00
torchscript_e2e_test Make error reporting a bit better. 2021-10-28 13:20:16 -07:00
CMakeLists.txt Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
annotations-sugar.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
lit.cfg.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
lit.site.cfg.py.in Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00