Prashant Kumar a3e4ea3228 Remove the dependency of the torchvision. (#858) ...

Remove the dependency of torchvision library for the conversion
of tensor layout format to what PIL library expects.

2023-01-23 08:49:57 -08:00

..

examples

Remove the dependency of the torchvision. (#858)

2023-01-23 08:49:57 -08:00

iree_utils

black format comments

2023-01-22 16:34:40 -08:00

sharkdynamo

Add sharkdynamo which combines shark with torchdynamo.

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

tests

Add stress test

2022-12-13 13:21:51 -08: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

[SharkInference] Make SharkInference compile the entire module (#708)

2023-01-03 23:25:23 +05:30

model_annotation.py

Add script to auto annotate SD models and variants (#751)

2023-01-04 15:53:10 -08:00

parser.py

Move most xfails to entries in tank/all_models.csv and temporarily remove multiprocessing and TF gpu support. (#646)

2022-12-16 12:56:32 +05:30

shark_benchmark_runner.py

Enable Torch-Inductor Part 2

2023-01-10 20:15:29 -08:00

shark_downloader.py

Revert "Change address for connection test (#785)" (#797)

2023-01-11 12:01:37 -08:00

shark_importer.py

[Shark][Training] Refresh SharkTrainer to latest APIs.

2023-01-19 20:30:15 +00:00

shark_inference.py

[SharkInference] Make SharkInference compile the entire module (#708)

2023-01-03 23:25:23 +05:30

shark_runner.py

[SharkInference] Make SharkInference compile the entire module (#708)

2023-01-03 23:25:23 +05:30

shark_trainer.py

[Shark][Training] Refresh SharkTrainer to latest APIs.

2023-01-19 20:30:15 +00:00

stress_test.py

Fix concurrency issue in stress_test for CUDA devices

2022-12-22 08:54:19 -08: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

Changes incorporating the recent torch_mlir compile api changes.

2022-11-15 15:25:37 +05:30