Commit Graph

12 Commits

Author SHA1 Message Date
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