Prashant Kumar
|
0dcf387089
|
Add shark_importer for torch_models. (#183)
All the torch_models are imported to gs::shark_tank.
Scripts have been updated.
|
2022-07-12 20:38:19 -07:00 |
|
Anush Elangovan
|
a7435973d9
|
Fix black formatting
|
2022-06-30 20:42:02 +00:00 |
|
Prashant Kumar
|
e8aa105b2a
|
Divide iree_utils and do module imports on function calls.
|
2022-06-22 14:17:33 +05:30 |
|
Prashant Kumar
|
4d3acff484
|
Add tf masked lm models (17 models - 12 pass, 5 fail). (#135)
|
2022-06-16 07:28:00 -07:00 |
|
Stanley Winata
|
c70ab55717
|
Add saving and loading of mlir to run. (#89)
|
2022-06-05 08:42:40 -07:00 |
|
Ean Garvey
|
8ad73d365c
|
Add support for training TF models + fix TF BERT training example (#80)
|
2022-06-01 18:46:07 -07:00 |
|
Prashant Kumar
|
cee02f6a61
|
Add frontend error checks.
|
2022-05-27 15:45:45 +05:30 |
|
Prashant Kumar
|
8294dc3f20
|
Bug fix related to torch-shark trainer.
|
2022-05-27 14:50:28 +05:30 |
|
Prashant Kumar
|
e5517f63f5
|
Add the refactored training module.
|
2022-05-27 14:21:34 +05:30 |
|
Prashant Kumar
|
18a4423026
|
Clearing the shark_trainer API for future rework.
|
2022-05-26 20:15:18 +05:30 |
|
Prashant Kumar
|
15c99adab8
|
Refactor the whole SharkInference API.
1. SharkInference API is refactored to target any frontends or files,
i.e. pytorch, tensorflow, mhlo, linalg, tosa.
2. Refactor all torch APIs.
|
2022-05-26 18:55:47 +05:30 |
|