llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/frontends/pytorch/python
History
Sean Silva 3d08c83580 Add flatten op recognition + shape refinement. 
This op has complex aliasing semantics, so it is kept mutable for now.

With this, we reduce ResNet18 to a single BB with all aten operators
having rank + dtype:
https://gist.github.com/silvasean/2fcb1c6e4d4ae27461204a43ae9c5031
 		2021-05-03 09:54:44 -07:00
..
torch_mlir Miscellaneous changes while trying to work on ResNet18 2021-04-27 11:51:11 -07:00
torch_mlir_utils/codegen Add flatten op recognition + shape refinement. 2021-05-03 09:54:44 -07:00
CMakeLists.txt Create public API for torch_mlir python code. 2020-10-13 16:36:49 -07:00