aceOfDiamonds commited on
Commit
0fbcd27
·
1 Parent(s): 3420a22

Nest the per-component methods

Browse files

The image and text towers each keep their own methods map now, matching the
schema's componentMethods.

Files changed (3) hide show
  1. coreml/config.json +49 -45
  2. vulkan/config.json +49 -45
  3. xnnpack/config.json +49 -45
coreml/config.json CHANGED
@@ -21,53 +21,57 @@
21
  "default": true,
22
  "methods": {
23
  "image": {
24
- "inputs": [
25
- {
26
- "shape": [
27
- 1,
28
- 3,
29
- 224,
30
- 224
31
- ],
32
- "dtype": "float32"
33
- }
34
- ],
35
- "outputs": [
36
- {
37
- "shape": [
38
- 1,
39
- 512
40
- ],
41
- "dtype": "float32"
42
- }
43
- ]
 
 
44
  },
45
  "text": {
46
- "inputs": [
47
- {
48
- "shape": [
49
- 1,
50
- 74
51
- ],
52
- "dtype": "int64"
53
- },
54
- {
55
- "shape": [
56
- 1,
57
- 74
58
- ],
59
- "dtype": "int64"
60
- }
61
- ],
62
- "outputs": [
63
- {
64
- "shape": [
65
- 1,
66
- 512
67
- ],
68
- "dtype": "float32"
69
- }
70
- ]
 
 
71
  }
72
  }
73
  }
 
21
  "default": true,
22
  "methods": {
23
  "image": {
24
+ "forward": {
25
+ "inputs": [
26
+ {
27
+ "shape": [
28
+ 1,
29
+ 3,
30
+ 224,
31
+ 224
32
+ ],
33
+ "dtype": "float32"
34
+ }
35
+ ],
36
+ "outputs": [
37
+ {
38
+ "shape": [
39
+ 1,
40
+ 512
41
+ ],
42
+ "dtype": "float32"
43
+ }
44
+ ]
45
+ }
46
  },
47
  "text": {
48
+ "forward": {
49
+ "inputs": [
50
+ {
51
+ "shape": [
52
+ 1,
53
+ 74
54
+ ],
55
+ "dtype": "int64"
56
+ },
57
+ {
58
+ "shape": [
59
+ 1,
60
+ 74
61
+ ],
62
+ "dtype": "int64"
63
+ }
64
+ ],
65
+ "outputs": [
66
+ {
67
+ "shape": [
68
+ 1,
69
+ 512
70
+ ],
71
+ "dtype": "float32"
72
+ }
73
+ ]
74
+ }
75
  }
76
  }
77
  }
vulkan/config.json CHANGED
@@ -21,53 +21,57 @@
21
  "default": true,
22
  "methods": {
23
  "image": {
24
- "inputs": [
25
- {
26
- "shape": [
27
- 1,
28
- 3,
29
- 224,
30
- 224
31
- ],
32
- "dtype": "float32"
33
- }
34
- ],
35
- "outputs": [
36
- {
37
- "shape": [
38
- 1,
39
- 512
40
- ],
41
- "dtype": "float32"
42
- }
43
- ]
 
 
44
  },
45
  "text": {
46
- "inputs": [
47
- {
48
- "shape": [
49
- 1,
50
- 74
51
- ],
52
- "dtype": "int64"
53
- },
54
- {
55
- "shape": [
56
- 1,
57
- 74
58
- ],
59
- "dtype": "int64"
60
- }
61
- ],
62
- "outputs": [
63
- {
64
- "shape": [
65
- 1,
66
- 512
67
- ],
68
- "dtype": "float32"
69
- }
70
- ]
 
 
71
  }
72
  }
73
  }
 
21
  "default": true,
22
  "methods": {
23
  "image": {
24
+ "forward": {
25
+ "inputs": [
26
+ {
27
+ "shape": [
28
+ 1,
29
+ 3,
30
+ 224,
31
+ 224
32
+ ],
33
+ "dtype": "float32"
34
+ }
35
+ ],
36
+ "outputs": [
37
+ {
38
+ "shape": [
39
+ 1,
40
+ 512
41
+ ],
42
+ "dtype": "float32"
43
+ }
44
+ ]
45
+ }
46
  },
47
  "text": {
48
+ "forward": {
49
+ "inputs": [
50
+ {
51
+ "shape": [
52
+ 1,
53
+ 74
54
+ ],
55
+ "dtype": "int64"
56
+ },
57
+ {
58
+ "shape": [
59
+ 1,
60
+ 74
61
+ ],
62
+ "dtype": "int64"
63
+ }
64
+ ],
65
+ "outputs": [
66
+ {
67
+ "shape": [
68
+ 1,
69
+ 512
70
+ ],
71
+ "dtype": "float32"
72
+ }
73
+ ]
74
+ }
75
  }
76
  }
77
  }
xnnpack/config.json CHANGED
@@ -16,53 +16,57 @@
16
  "default": true,
17
  "methods": {
18
  "image": {
19
- "inputs": [
20
- {
21
- "shape": [
22
- 1,
23
- 3,
24
- 224,
25
- 224
26
- ],
27
- "dtype": "float32"
28
- }
29
- ],
30
- "outputs": [
31
- {
32
- "shape": [
33
- 1,
34
- 512
35
- ],
36
- "dtype": "float32"
37
- }
38
- ]
 
 
39
  },
40
  "text": {
41
- "inputs": [
42
- {
43
- "shape": [
44
- 1,
45
- 74
46
- ],
47
- "dtype": "int64"
48
- },
49
- {
50
- "shape": [
51
- 1,
52
- 74
53
- ],
54
- "dtype": "int64"
55
- }
56
- ],
57
- "outputs": [
58
- {
59
- "shape": [
60
- 1,
61
- 512
62
- ],
63
- "dtype": "float32"
64
- }
65
- ]
 
 
66
  }
67
  },
68
  "components": {
 
16
  "default": true,
17
  "methods": {
18
  "image": {
19
+ "forward": {
20
+ "inputs": [
21
+ {
22
+ "shape": [
23
+ 1,
24
+ 3,
25
+ 224,
26
+ 224
27
+ ],
28
+ "dtype": "float32"
29
+ }
30
+ ],
31
+ "outputs": [
32
+ {
33
+ "shape": [
34
+ 1,
35
+ 512
36
+ ],
37
+ "dtype": "float32"
38
+ }
39
+ ]
40
+ }
41
  },
42
  "text": {
43
+ "forward": {
44
+ "inputs": [
45
+ {
46
+ "shape": [
47
+ 1,
48
+ 74
49
+ ],
50
+ "dtype": "int64"
51
+ },
52
+ {
53
+ "shape": [
54
+ 1,
55
+ 74
56
+ ],
57
+ "dtype": "int64"
58
+ }
59
+ ],
60
+ "outputs": [
61
+ {
62
+ "shape": [
63
+ 1,
64
+ 512
65
+ ],
66
+ "dtype": "float32"
67
+ }
68
+ ]
69
+ }
70
  }
71
  },
72
  "components": {