Ean Garvey 08eda2ce35 Equip TF tests with save_mlir, save_vmfb, and benchmark to .csv options. (#148)

2022-06-21 21:54:44 -07:00

..

examples

Changes to minilm to make it one on one with tensorflow.

2022-06-21 22:08:59 +05:30

tests

Refactor testing framework. (#117)

2022-06-12 20:15:30 -07:00

__init__.py

Modify the directory structure to remove conflicts.

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

backward_makefx.py

Add tf masked lm models (17 models - 12 pass, 5 fail). (#135)

2022-06-16 07:28:00 -07:00

cuda_utils.py

Add tf masked lm models (17 models - 12 pass, 5 fail). (#135)

2022-06-16 07:28:00 -07:00

iree_eager_backend.py

Refactor the whole SharkInference API.

2022-05-26 18:55:47 +05:30

iree_utils.py

Equip TF tests with save_mlir, save_vmfb, and benchmark to .csv options. (#148)

2022-06-21 21:54:44 -07:00

model_annotation.py

yapf format python files.

2022-06-06 14:34:59 +05:30

parser.py

yapf format python files.

2022-06-06 14:34:59 +05:30

shark_downloader.py

Add SharkDownloader for end users (#147)

2022-06-21 16:42:23 -07:00

shark_importer.py

Enable tosa.mlir as input for SharkImporter inference (#145)

2022-06-20 23:15:14 -07:00

shark_inference.py

Equip TF tests with save_mlir, save_vmfb, and benchmark to .csv options. (#148)

2022-06-21 21:54:44 -07:00

shark_runner.py

Equip TF tests with save_mlir, save_vmfb, and benchmark to .csv options. (#148)

2022-06-21 21:54:44 -07:00

shark_trainer.py

Add tf masked lm models (17 models - 12 pass, 5 fail). (#135)

2022-06-16 07:28:00 -07:00

torch_mlir_utils.py

Add tf masked lm models (17 models - 12 pass, 5 fail). (#135)

2022-06-16 07:28:00 -07:00