Ean Garvey c5ae01311d Make PyTorch model tests importer-independent and add benchmarks.

2022-07-15 21:54:41 -05:00

..

__init__.py

Divide iree_utils and do module imports on function calls.

2022-06-22 14:17:33 +05:30

_common.py

Update pytorch tests to support vulkan and cuda.

2022-07-08 14:40:13 +05:30

benchmark_utils.py

Make PyTorch model tests importer-independent and add benchmarks.

2022-07-15 21:54:41 -05:00

compile_utils.py

Add shark_importer for torch_models. (#183)

2022-07-12 20:38:19 -07:00

cpu_utils.py

Add shark_importer for torch_models. (#183)

2022-07-12 20:38:19 -07:00

gpu_utils.py

Add shark_importer for torch_models. (#183)

2022-07-12 20:38:19 -07:00

vulkan_utils.py

Bug fix: Pass the device attribute appropriately.

2022-07-09 01:56:17 +05:30