Prashant Kumar 315ec72984 Add check for hash.npy in the models folder.

2022-08-01 22:40:25 +05:30

..

examples

Enable --devices to pass through CLI.

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

iree_utils

Make PyTorch model tests importer-independent and add benchmarks.

2022-07-15 21:54:41 -05: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

Add shark_importer for torch_models. (#183)

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

model_annotation.py

Add shark_importer for torch_models. (#183)

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

parser.py

Add shark_importer for torch_models. (#183)

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

shark_benchmark_runner.py

Fix GPU benchmarks on PyTorch model tests. (#218)

2022-07-29 15:17:55 -07:00

shark_downloader.py

Add check for hash.npy in the models folder.

2022-08-01 22:40:25 +05:30

shark_importer.py

Add tf image classification auto model (#213)

2022-07-26 23:18:42 -07: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

Add shark_importer for torch_models. (#183)

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