dan 6eea111f5e bert training wip

2022-11-17 17:24:46 +00:00

..

examples

bert training wip

2022-11-17 17:24:46 +00:00

iree_utils

Fix issues with loading .vmfb into SharkInference

2022-09-23 09:53:13 +05:30

sharkdynamo

Add sharkdynamo which combines shark with torchdynamo.

2022-09-23 22:40:02 +05:30

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

bert training wip

2022-11-17 17:24:46 +00:00

iree_eager_backend.py

Fix IREE eager backend device string (#237)

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

model_annotation.py

Update model annotation tool (#361)

2022-09-23 15:44:51 -07:00

parser.py

Add --local_tank_cache flag and update requirements. (#368)

2022-09-28 03:02:59 -05:00

shark_benchmark_runner.py

creates abstract test case class (#333)

2022-09-20 07:06:38 -07:00

shark_downloader.py

Add --local_tank_cache flag and update requirements. (#368)

2022-09-28 03:02:59 -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

Fix issues with loading .vmfb into SharkInference

2022-09-23 09:53:13 +05:30

shark_runner.py

bert training wip

2022-11-17 17:24:46 +00:00

shark_trainer.py

bert training wip

2022-11-17 17:24:46 +00:00

tflite_utils.py

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

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

torch_mlir_lockstep_tensor.py

[Lockstep] Hack to avoid aten._reshape_alias (#332)

2022-09-22 18:02:09 -04:00

torch_mlir_utils.py

Fix tmp dir bug (#285)

2022-08-22 15:00:35 -07:00