21 Commits

Author SHA1 Message Date
Ean Garvey
a504903dd5 Fix formatting issues. (#903) 2023-02-01 09:12:45 +05:30
Ean Garvey
3618fb3ada Move old test scripts out of base tank directory and add xfails. (#389) 2022-10-07 16:02:46 -07:00
Chi_Liu
dc1a283ab7 Remove tf package dep for SharkDownloader tflite tests (#212) 2022-07-26 12:41:12 -07:00
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
Chi_Liu
1cad50d521 generate shark tank for tflite (#173)
* Add gen_shark_tank support tflite

* gen_shark_tank.py use SharkImporter to save model
2022-07-05 23:11:19 -07:00
Chi Liu
193a735865 Move tflite examples to up dir 2022-06-30 14:57:33 -07:00
Anush Elangovan
a7435973d9 Fix black formatting 2022-06-30 20:42:02 +00:00
Chi_Liu
2adea76b8c Enable SharkImporter Tests (#165)
rewrite some tflite code from SharkImporter to tflite_utils.py
2022-06-29 10:11:48 -07:00
Chi_Liu
210d28fce9 Add more tflite shark_importer examples (#163) 2022-06-28 08:39:56 -07:00
Chi_Liu
44dce561e9 SharkImporter for tflite without forward and compile (#159) 2022-06-23 22:49:35 -07:00
Chi_Liu
fa0aaf63c2 Fix repo_dir cannot export/write mlir file bug (#157)
-Set repro_dir as a temporary dir within current working directory.
2022-06-23 18:37:16 -07:00
Chi_Liu
a635b6fbef Add tf/torch/mhlo/tosa support for SharkDownloader (#151) 2022-06-22 11:25:34 -07:00
Prashant Kumar
e8aa105b2a Divide iree_utils and do module imports on function calls. 2022-06-22 14:17:33 +05:30
Ean Garvey
08eda2ce35 Equip TF tests with save_mlir, save_vmfb, and benchmark to .csv options. (#148) 2022-06-21 21:54:44 -07:00
Chi_Liu
206ba11cc8 Add SharkDownloader for end users (#147)
* Add SharkDownloader for user

* Change tank_url to gs://shark_tank
2022-06-21 16:42:23 -07:00
Chi_Liu
af582925f2 Enable tosa.mlir as input for SharkImporter inference (#145)
* Change shark_importer to use tosa.mlir as tflite model input from  local gen_shark_tank
2022-06-20 23:15:14 -07:00
Chi_Liu
f6e9f2d571 Generate Shark Tank for tflite (#142)
generate shark_tank for tflite
2022-06-20 14:35:35 -07:00
Prashant Kumar
4d3acff484 Add tf masked lm models (17 models - 12 pass, 5 fail). (#135) 2022-06-16 07:28:00 -07:00
Chi_Liu
8640c3ebfe Add shark_importer tflite module and albert_shark_test example (#110) 2022-06-09 06:17:22 -07:00
powderluv
2b21c8d19d Delete .albert_test.py.swp 2022-06-06 23:45:22 -07:00
anush elangovan
1fa05fc7c8 Rename to tflite and ignore lit.cfg.py in flake8 2022-06-03 05:21:23 +00:00