73 Commits

Author	SHA1	Message	Date
Ean Garvey	b522f933a9	add/remove pytest marks to TF tests	2022-06-20 15:23:28 +00:00
Ean Garvey	dbabb48582	Add TempFileSaver pytest option to tank TF tests.	2022-06-20 15:02:36 +00:00
Anush Elangovan	2b3fac7519	Vulkan xfails ... https://github.com/google/iree/issues/9554	2022-06-18 09:39:09 +00:00
Anush Elangovan	de223e7824	Add xfail reason	2022-06-18 08:31:26 +00:00
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