torch-mlir/python
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
..
test Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
torch_mlir/dialects Add BertSequenceClassification model to e2e 2021-09-30 13:30:29 -04:00
torch_mlir_e2e_test 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
TorchMLIRModule.cpp Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00