olive-recipes/google-vit-base-patch16-224/aitk/info.yml at main · CodeLinaro/olive-recipes · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
keywords:
    aitk
arch: vit
recipes:
    - file: "vit-base-patch16-224_qdq_qnn.json"
      device: npu
      ep: QNNExecutionProvider
    - file: "vit-base-patch16-224_qdq_amd.json"
      device: npu
      ep: VitisAIExecutionProvider
    - file: "vit_base_patch16_224_context_ov_static.json"
      devices:
        - npu
        - cpu
        - gpu
      ep: OpenVINOExecutionProvider
    - file: "vit-base-patch16-224_trtrtx.json"
      device: gpu
      ep: NvTensorRTRTXExecutionProvider
    - file: "vit-base-patch16-224_dml.json"
      device: gpu
      ep: DmlExecutionProvider
    - file: "vit-base-patch16-224_migraphx.json"
      device: gpu
      ep: MIGraphXExecutionProvider
    - file: "vit-base-patch16-224_qnn_gpu.json"
      device: gpu
      ep: QNNExecutionProvider
aitk:
    modelInfo:
        id: "huggingface/google/vit-base-patch16-224"
        version: 5
        p0: true