Vivek Khandelwal
|
11f62d7fac
|
Minor fixes for MiniLM Training
|
2023-07-18 17:16:44 +05:30 |
|
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 |
|
Prashant Kumar
|
e5517f63f5
|
Add the refactored training module.
|
2022-05-27 14:21:34 +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 |
|
Prashant Kumar
|
6a358cb90a
|
Update the examples removing functorch dependency.
|
2022-05-24 19:10:01 +05:30 |
|
Prashant Kumar
|
96577000a3
|
Remove functorch dependency.
|
2022-05-24 18:42:35 +05:30 |
|
Prashant Kumar
|
a280b0a651
|
Add decompositions for bert_training.
|
2022-05-09 16:39:11 +05:30 |
|
Prashant Kumar
|
b4114c0938
|
Make backward API workable.
|
2022-05-06 14:51:44 +05:30 |
|
Prashant Kumar
|
08df345fbc
|
Add make_fx whole graph extract.
|
2022-05-04 20:58:59 +05:30 |
|