6 Commits

Author	SHA1	Message	Date
yzhang93	587d74b449	Update model annotation tool (#361) ... Usage: with create_context() as ctx: module = model_annotation(ctx, input_contents=..., config_path=..., search_op=...) Example: The example is to annotate the minilm model with GPU config files. python model_annotation.py /nodclouddata/vivian/minilm_model/model.mlir /nodclouddata/vivian/minilm_model/model_config.json	2022-09-23 15:44:51 -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
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	9d2ce33bd1	yapf format python files. ... Please use `yapf -i --style .style.yapf shark/*.py` to format python files.	2022-06-06 14:34:59 +05:30
yzhang93	16c50cac15	Add the option to use tuned model in shark_runner (#79)	2022-06-02 18:25:33 -07:00