msluszniak commited on
Commit
8e8a387
·
verified ·
1 Parent(s): 729b5cf

Add backend config (ExecuTorch 1.4.1)

Browse files
Files changed (1) hide show
  1. coreml/config.json +140 -0
coreml/config.json ADDED
@@ -0,0 +1,140 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "$schema": "https://huggingface.co/software-mansion/react-native-executorch-spec/resolve/main/config.schema.json",
3
+ "model": "selfie_segmentation",
4
+ "family": "selfie",
5
+ "capabilities": [
6
+ "semantic-segmentation"
7
+ ],
8
+ "backend": "coreml",
9
+ "license": "apache-2.0",
10
+ "variants": [
11
+ {
12
+ "file": "selfie_segmentation_coreml_fp16.pte",
13
+ "precision": "fp16",
14
+ "quantized": false,
15
+ "default": true,
16
+ "methods": {
17
+ "forward": {
18
+ "inputs": [
19
+ {
20
+ "shape": [
21
+ 1,
22
+ 3,
23
+ 256,
24
+ 256
25
+ ],
26
+ "dtype": "float32"
27
+ }
28
+ ],
29
+ "outputs": [
30
+ {
31
+ "shape": [
32
+ 1,
33
+ 1,
34
+ 256,
35
+ 256
36
+ ],
37
+ "dtype": "float32"
38
+ }
39
+ ]
40
+ }
41
+ }
42
+ },
43
+ {
44
+ "file": "selfie_segmentation_landscape_coreml_fp16.pte",
45
+ "precision": "fp16",
46
+ "quantized": false,
47
+ "default": false,
48
+ "methods": {
49
+ "forward": {
50
+ "inputs": [
51
+ {
52
+ "shape": [
53
+ 1,
54
+ 3,
55
+ 144,
56
+ 256
57
+ ],
58
+ "dtype": "float32"
59
+ }
60
+ ],
61
+ "outputs": [
62
+ {
63
+ "shape": [
64
+ 1,
65
+ 1,
66
+ 144,
67
+ 256
68
+ ],
69
+ "dtype": "float32"
70
+ }
71
+ ]
72
+ }
73
+ }
74
+ },
75
+ {
76
+ "file": "selfie_segmentation_coreml_fp32.pte",
77
+ "precision": "fp32",
78
+ "quantized": false,
79
+ "default": false,
80
+ "methods": {
81
+ "forward": {
82
+ "inputs": [
83
+ {
84
+ "shape": [
85
+ 1,
86
+ 3,
87
+ 256,
88
+ 256
89
+ ],
90
+ "dtype": "float32"
91
+ }
92
+ ],
93
+ "outputs": [
94
+ {
95
+ "shape": [
96
+ 1,
97
+ 1,
98
+ 256,
99
+ 256
100
+ ],
101
+ "dtype": "float32"
102
+ }
103
+ ]
104
+ }
105
+ }
106
+ },
107
+ {
108
+ "file": "selfie_segmentation_landscape_coreml_fp32.pte",
109
+ "precision": "fp32",
110
+ "quantized": false,
111
+ "default": false,
112
+ "methods": {
113
+ "forward": {
114
+ "inputs": [
115
+ {
116
+ "shape": [
117
+ 1,
118
+ 3,
119
+ 144,
120
+ 256
121
+ ],
122
+ "dtype": "float32"
123
+ }
124
+ ],
125
+ "outputs": [
126
+ {
127
+ "shape": [
128
+ 1,
129
+ 1,
130
+ 144,
131
+ 256
132
+ ],
133
+ "dtype": "float32"
134
+ }
135
+ ]
136
+ }
137
+ }
138
+ }
139
+ ]
140
+ }