Chi_Liu
0c31bb82cd
Add hash of mlir for tf/tflite ( #225 )
2022-08-01 10:40:49 -07:00
Chi_Liu
c1cb7bb3fd
Rename tank tflite/torch model dir ( #219 )
2022-07-29 01:56:05 -05:00
Prashant Kumar
ff20ddeb97
Add hash of the mlir for checking upstream changes. ( #217 )
...
-- hash of the mlir files have been added in the shark_tank. (If there
is upstream changes, the model will be downloaded otherwise not).
-- The models have been placed in `~/.local/shark_tank/` folder as
opposed to the current folder not to redownload the files again.
2022-07-28 21:54:50 -05:00
Chi_Liu
2c7d879a4e
Add tf image classification auto model tests ( #216 )
2022-07-27 18:38:13 -07:00
Chi_Liu
af4257d05f
Add tf image classification auto model ( #213 )
2022-07-26 23:18:42 -07:00
Chi_Liu
dc1a283ab7
Remove tf package dep for SharkDownloader tflite tests ( #212 )
2022-07-26 12:41:12 -07:00
Prashant Kumar
49fc6d2f4b
Add all the tf models to gs::shark_tank. ( #207 )
...
-- Tensorflow models have been imported to shark_tank.
-- TF Tests have been updated to use the downloader.
2022-07-22 11:53:07 -05:00
Prashant Kumar
e62c531170
Update the torch_model for dynamic cases.
2022-07-22 00:21:35 +05:30
Prashant Kumar
3c2f8ef243
Generate shark_tank for tensorflow models.
...
Updated the generate_sharktank.py script to give tensorflow models.
2022-07-20 22:37:14 +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
Chi_Liu
b98c9ee484
torch fix in gen_shark_tank ( #180 )
2022-07-08 10:13:58 -07:00
Chi_Liu
d496c66d11
Torch support of gen_shark_tank ( #175 )
2022-07-06 22:42:15 -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
Anush Elangovan
a7435973d9
Fix black formatting
2022-06-30 20:42:02 +00:00
Chi_Liu
44dce561e9
SharkImporter for tflite without forward and compile ( #159 )
2022-06-23 22:49:35 -07:00
Prashant Kumar
e8aa105b2a
Divide iree_utils and do module imports on function calls.
2022-06-22 14:17:33 +05:30
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
