219 Commits

Author SHA1 Message Date
Ean Garvey
ce7350f5f8 Mark all TF model tests as expected to fail. (#139) 2022-06-16 16:25:18 -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
Prashant Kumar
b43d627b20 Add distilbert-base-uncased to tank pytorch test.
distilbert-base-uncased model validation test is added.
2022-06-15 21:14:56 +05:30
Ean Garvey
dc8c221467 Add option to save .vmfb in pytest + move pytorch model tests. (#131) 2022-06-14 20:39:10 -07:00
Prashant Kumar
380b0b7c54 Add seq_classification shark module to run seq_classification
on hugging face supported models.
2022-06-14 22:36:19 +05:30
Ean Garvey
01f900a9ff Move conftest.py to fix nightly release. (#126)
* Move conftest.py into frontend directory.

* xfail albert static on vulkan backend.
2022-06-14 07:07:33 -07:00
Ean Garvey
ec13200c7c Update conftest.py 2022-06-13 19:51:21 -05:00
Ean Garvey
e872296498 Enable passing of --save-mlir cmd-line option through pytest. 2022-06-14 00:27:34 +00:00
Ean Garvey
acdd1d1785 Update model testing utilities. (#119) 2022-06-13 02:16:30 -07:00
Anush Elangovan
3ea34d08e2 Add __init__.py for tank/ 2022-06-12 21:53:59 -07:00
Ean Garvey
cbd131d588 Refactor testing framework. (#117) 2022-06-12 20:15:30 -07:00
Prashant Kumar
93bf13199f Add tensorflow models under AutoModelForMaskedLM class. 2022-06-09 22:02:21 +05:30
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
Maksim Levental
1eb493dae9 Demo for currently not MLIR lowerable v-diffusion model (including both demo path and traceable path). (#86) 2022-06-03 14:17:50 -07:00
anush elangovan
9c70dea957 Pull in TF models 2022-06-03 06:10:24 +00:00
anush elangovan
1fa05fc7c8 Rename to tflite and ignore lit.cfg.py in flake8 2022-06-03 05:21:23 +00:00
anush elangovan
c6ea400178 Fix lit flake8 errors 2022-06-02 18:56:55 +00:00
anush elangovan
395adc2921 Add tflitehub 2022-06-02 16:58:18 +00:00