Skip to content

Commit 0322941

Browse files
committed
Address review comments
1 parent 8aa7965 commit 0322941

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed
4 Bytes
Binary file not shown.

onnxruntime/test/testdata/test_evil_weights.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,6 @@ def create_exp_model():
2424
entry1.value = "*/_ORT_MEM_ADDR_/*"
2525
entry2 = evil_tensor.external_data.add()
2626
entry2.key = "offset"
27-
entry2.value = "4194304"
2827
entry2.value = "12230656"
2928
entry3 = evil_tensor.external_data.add()
3029
entry3.key = "length"
@@ -33,7 +32,7 @@ def create_exp_model():
3332
tensors.append(onnx.helper.make_tensor(name="0x1", data_type=onnx.TensorProto.INT64, dims=[1], vals=[0x1]))
3433
nodes.append(onnx.helper.make_node(op_type="Add", inputs=["evil_weights", "0x1"], outputs=["output"]))
3534

36-
outputs.append(onnx.helper.make_tensor_value_info("output", onnx.TensorProto.INT64, []))
35+
outputs.append(onnx.helper.make_tensor_value_info("output", onnx.TensorProto.INT64, [10]))
3736

3837
graph = onnx.helper.make_graph(nodes, "test", inputs, outputs, tensors)
3938
model = onnx.helper.make_model(

0 commit comments

Comments
 (0)