Boian Petkantchin eba4d06405 In MiniLM JAX example do not hardcode device (#1385) ...

* In MiniLM JAX example do not hardcode device

* In MiniLM JAX example don't use bytecode MLIR

---------

Co-authored-by: Boian Petkantchin <boian@nod-labs.com>

2023-05-03 10:34:42 -07:00

..

ESRGAN

Fix formatting issues. (#903)

2023-02-01 09:12:45 +05:30

llama

Update README.md

2023-03-12 16:54:42 -07:00

upscaler

Revert "Revert pipelines (#948)" (#951)

2023-02-04 22:44:08 -05:00

albert_maskfill_pt.py

Level Zero Backend (#280)

2022-08-17 19:19:27 -07:00

albert_maskfill_tf.py

Enable XLA compiler for TF models (#484)

2022-11-13 20:10:47 -08:00

bloom_tank.py

Revert "Revert "find gsutil on linux (#557)" (#560)" (#561)

2022-12-05 20:27:43 -08:00

CLIPModel_tf.py

Enable XLA compiler for TF models (#484)

2022-11-13 20:10:47 -08:00

gpt2_tf.py

Enable XLA compiler for TF models (#484)

2022-11-13 20:10:47 -08:00

mhlo_example.py

Add shark_importer for torch_models. (#183)

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

minilm_benchmark_tf.py

Enable XLA compiler for TF models (#484)

2022-11-13 20:10:47 -08:00

minilm_benchmark.py

Divide iree_utils and do module imports on function calls.

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

minilm_jax_requirements.txt

In MiniLM JAX example do not hardcode device (#1385)

2023-05-03 10:34:42 -07:00

minilm_jax.py

In MiniLM JAX example do not hardcode device (#1385)

2023-05-03 10:34:42 -07:00

minilm_jit.py

Revert "Revert "find gsutil on linux (#557)" (#560)" (#561)

2022-12-05 20:27:43 -08:00

minilm_tf_gpu_config.json

Add the option to use tuned model in shark_runner (#79)

2022-06-02 18:25:33 -07:00

minilm_tf.py

Enable XLA compiler for TF models (#484)

2022-11-13 20:10:47 -08:00

resnest.py

Update unet_model to run on shark.

2022-08-17 13:16:02 +05:30

resnet50_fp16.py

Add FP16 Resnet50 script

2022-10-06 21:56:43 +05:30

resnet50_script.py

Revert "Revert "find gsutil on linux (#557)" (#560)" (#561)

2022-12-05 20:27:43 -08:00

sharded_bloom_large_models.py

Fix bloom lint

2023-03-10 11:53:08 -08:00

sharded_bloom.py

Fix bloom lint

2023-03-10 11:53:08 -08:00

simple_dlrm.py

Fix formatting issues. (#903)

2023-02-01 09:12:45 +05:30

sparse_arch.py

Fix formatting issues. (#903)

2023-02-01 09:12:45 +05:30

t5_tf.py

Enable XLA compiler for TF models (#484)

2022-11-13 20:10:47 -08:00

torch_vision_models_script.py

Divide iree_utils and do module imports on function calls.

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

unet_script.py

Update unet_model to run on shark.

2022-08-17 13:16:02 +05:30

v_diffusion.py

Revert "Revert "find gsutil on linux (#557)" (#560)" (#561)

2022-12-05 20:27:43 -08:00