Prashant Kumar d10d6e35ab Add distilbert model to the tests.

2022-06-06 19:48:29 +05:30

..

examples

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

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

tests

Add distilbert model to the tests.

2022-06-06 19:48:29 +05:30

__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

yapf format python files.

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

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_inference.py

yapf format python files.

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

shark_runner.py

yapf format python files.

2022-06-06 14:34:59 +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