torch-mlir/lib/Dialect/Torch
Yi Zhang 89225b0cd8 Add BertSequenceClassification model to e2e
Use torch tracing to get the module because the original model is not
TorchScriptable out of box.
2021-09-30 13:30:29 -04:00
..
IR Add BertSequenceClassification model to e2e 2021-09-30 13:30:29 -04:00
Transforms Implement the lazytensor package (#331) 2021-09-28 17:25:06 -07:00
CMakeLists.txt Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00