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/compile_api
History
Sean Silva 31727f81d8 torch_mlir.compile: Allow ignoring traced shapes 
In some cases, users know that a traced graph is valid for a wider set
of shapes than they originally traced it with. Provide an option for
users to ignore the shapes in the traced graph when they know it is
legal.

Fixes #997
 		2022-08-04 10:18:34 -07:00
..
basic.py Update test to pass with newer versions of tanh (#990) 2022-06-28 20:28:13 -07:00
output_type_spec.py torch_mlir.compile: Allow OutputType as a string. 2022-07-08 17:37:27 -07:00
tracing.py torch_mlir.compile: Allow ignoring traced shapes 2022-08-04 10:18:34 -07:00