Ean Garvey a3654f33da Fix sourcing for canonical MiniLM shark_tank model artifacts. (#278) ...

* Fix generation of MiniLM artifacts.

* Fix miniLM output for validation. Xfail numerics failure on mpnet.

* Update distilbert-base-uncased_tf_test.py

* try-except for transition of minilm model

2022-08-17 23:03:47 -05:00

..

examples

Level Zero Backend (#280)

2022-08-17 19:19:27 -07:00

iree_utils

Level Zero Backend (#280)

2022-08-17 19:19:27 -07:00

tests

Remove tf package dep for SharkDownloader tflite tests (#212)

2022-07-26 12:41:12 -07:00

__init__.py

Modify the directory structure to remove conflicts.

2022-04-04 21:23:33 +05:30

backward_makefx.py

Add shark_importer for torch_models. (#183)

2022-07-12 20:38:19 -07:00

iree_eager_backend.py

Fix IREE eager backend device string (#237)

2022-08-03 12:09:52 -07:00

model_annotation.py

Add shark_importer for torch_models. (#183)

2022-07-12 20:38:19 -07:00

parser.py

Improvements to pytest benchmarks. (#267)

2022-08-17 02:29:48 -05:00

shark_benchmark_runner.py

Improvements to pytest benchmarks. (#267)

2022-08-17 02:29:48 -05:00

shark_downloader.py

fix missing model download path (#281)

2022-08-17 23:02:50 -05:00

shark_importer.py

Fix sourcing for canonical MiniLM shark_tank model artifacts. (#278)

2022-08-17 23:03:47 -05:00

shark_inference.py

Enable --devices to pass through CLI.

2022-07-27 22:13:52 +05:30

shark_runner.py

Enable --devices to pass through CLI.

2022-07-27 22:13:52 +05:30

shark_trainer.py

Add shark_importer for torch_models. (#183)

2022-07-12 20:38:19 -07:00

tflite_utils.py

Remove tf package dep for SharkDownloader tflite tests (#212)

2022-07-26 12:41:12 -07:00

torch_mlir_utils.py

Update torch_mlir.compile API.

2022-08-10 22:50:15 +05:30