stanley 14a56ca9b0 Mini LM Loader Example ...

-Add example to load miniLM from SharkHUB and benchmark.
-Modify TF benchmark to have growing GPU allocation.
-Add shark_load helper function

2022-06-15 02:57:42 +00:00

..

examples

Mini LM Loader Example

2022-06-15 02:57:42 +00:00

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 the refactored training module.

2022-05-27 14:21:34 +05:30

cuda_utils.py

yapf format python files.

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

iree_eager_backend.py

Refactor the whole SharkInference API.

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

iree_utils.py

Mini LM Loader Example

2022-06-15 02:57:42 +00: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_importer.py

Mini LM Loader Example

2022-06-15 02:57:42 +00:00

shark_inference.py

Mini LM Loader Example

2022-06-15 02:57:42 +00:00

shark_runner.py

Add seq_classification shark module to run seq_classification

2022-06-14 22:36:19 +05:30

shark_trainer.py

Add saving and loading of mlir to run. (#89)

2022-06-05 08:42:40 -07:00

torch_mlir_utils.py

Add saving and loading of mlir to run. (#89)

2022-06-05 08:42:40 -07:00