Skip to content

Commit 217dbab

Browse files
wschinshauheen
authored andcommitted
Fix another model file's opset_import (#7)
* Fix another model file's opset_import * Use meaningful graph name rather than GUID
1 parent 59d2f56 commit 217dbab

File tree

3 files changed

+42
-34
lines changed

3 files changed

+42
-34
lines changed

tests/baseline/models/keras2coreml_Dense_ImageNet_small.json

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ graph {
1818
ints: 1
1919
type: INTS
2020
}
21+
domain: ""
2122
}
2223
node {
2324
input: "Transpose"
@@ -29,6 +30,7 @@ graph {
2930
i: 1
3031
type: INT
3132
}
33+
domain: ""
3234
}
3335
node {
3436
input: "flatten_1_output"
@@ -47,8 +49,9 @@ graph {
4749
i: 1
4850
type: INT
4951
}
52+
domain: ""
5053
}
51-
name: "1473999ee74f4db1b7d88c2dc6a64a0d"
54+
name: "keras2coreml_Dense_ImageNet_small"
5255
initializer {
5356
dims: 2
5457
dims: 27
@@ -155,3 +158,8 @@ graph {
155158
}
156159
}
157160
}
161+
opset_import {
162+
domain: ""
163+
version: 1
164+
}
165+

tests/utils/models/coreml_OneHotEncoder_BikeSharing.json

Lines changed: 33 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -11,13 +11,13 @@ graph {
1111
name: "Scaler"
1212
op_type: "Scaler"
1313
attribute {
14-
name: "offset"
15-
floats: 0.0
14+
name: "scale"
15+
floats: 1.0
1616
type: FLOATS
1717
}
1818
attribute {
19-
name: "scale"
20-
floats: 1.0
19+
name: "offset"
20+
floats: 0.0
2121
type: FLOATS
2222
}
2323
domain: "ai.onnx.ml"
@@ -28,13 +28,13 @@ graph {
2828
name: "Scaler.1"
2929
op_type: "Scaler"
3030
attribute {
31-
name: "offset"
32-
floats: 0.0
31+
name: "scale"
32+
floats: 1.0
3333
type: FLOATS
3434
}
3535
attribute {
36-
name: "scale"
37-
floats: 1.0
36+
name: "offset"
37+
floats: 0.0
3838
type: FLOATS
3939
}
4040
domain: "ai.onnx.ml"
@@ -45,13 +45,13 @@ graph {
4545
name: "Scaler.2"
4646
op_type: "Scaler"
4747
attribute {
48-
name: "offset"
49-
floats: 0.0
48+
name: "scale"
49+
floats: 1.0
5050
type: FLOATS
5151
}
5252
attribute {
53-
name: "scale"
54-
floats: 1.0
53+
name: "offset"
54+
floats: 0.0
5555
type: FLOATS
5656
}
5757
domain: "ai.onnx.ml"
@@ -62,13 +62,13 @@ graph {
6262
name: "Scaler.3"
6363
op_type: "Scaler"
6464
attribute {
65-
name: "offset"
66-
floats: 0.0
65+
name: "scale"
66+
floats: 1.0
6767
type: FLOATS
6868
}
6969
attribute {
70-
name: "scale"
71-
floats: 1.0
70+
name: "offset"
71+
floats: 0.0
7272
type: FLOATS
7373
}
7474
domain: "ai.onnx.ml"
@@ -79,13 +79,13 @@ graph {
7979
name: "Scaler.4"
8080
op_type: "Scaler"
8181
attribute {
82-
name: "offset"
83-
floats: 0.0
82+
name: "scale"
83+
floats: 1.0
8484
type: FLOATS
8585
}
8686
attribute {
87-
name: "scale"
88-
floats: 1.0
87+
name: "offset"
88+
floats: 0.0
8989
type: FLOATS
9090
}
9191
domain: "ai.onnx.ml"
@@ -96,13 +96,13 @@ graph {
9696
name: "Scaler.5"
9797
op_type: "Scaler"
9898
attribute {
99-
name: "offset"
100-
floats: 0.0
99+
name: "scale"
100+
floats: 1.0
101101
type: FLOATS
102102
}
103103
attribute {
104-
name: "scale"
105-
floats: 1.0
104+
name: "offset"
105+
floats: 0.0
106106
type: FLOATS
107107
}
108108
domain: "ai.onnx.ml"
@@ -113,13 +113,13 @@ graph {
113113
name: "Scaler.6"
114114
op_type: "Scaler"
115115
attribute {
116-
name: "offset"
117-
floats: 0.0
116+
name: "scale"
117+
floats: 1.0
118118
type: FLOATS
119119
}
120120
attribute {
121-
name: "scale"
122-
floats: 1.0
121+
name: "offset"
122+
floats: 0.0
123123
type: FLOATS
124124
}
125125
domain: "ai.onnx.ml"
@@ -130,13 +130,13 @@ graph {
130130
name: "Scaler.7"
131131
op_type: "Scaler"
132132
attribute {
133-
name: "offset"
134-
floats: 0.0
133+
name: "scale"
134+
floats: 1.0
135135
type: FLOATS
136136
}
137137
attribute {
138-
name: "scale"
139-
floats: 1.0
138+
name: "offset"
139+
floats: 0.0
140140
type: FLOATS
141141
}
142142
domain: "ai.onnx.ml"
@@ -402,7 +402,7 @@ graph {
402402
}
403403
domain: "ai.onnx.ml"
404404
}
405-
name: "e29a78efcc8341a992cfc72e0e6f5260"
405+
name: "coreml_OneHotEncoder_BikeSharing"
406406
initializer {
407407
dims: 1
408408
data_type: INT64
0 Bytes
Binary file not shown.

0 commit comments

Comments
 (0)