Skip to content

Commit

Permalink
add expand conformance tests (#1098)
Browse files Browse the repository at this point in the history
* add expand conformance tests

* fix clip-vit-base-head model and its output
  • Loading branch information
fengyuentau committed Sep 27, 2023
1 parent 1aadd31 commit 6608980
Show file tree
Hide file tree
Showing 21 changed files with 1 addition and 1 deletion.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file modified testdata/dnn/onnx/data/output_clip-vit-base-head.npy
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
2 changes: 1 addition & 1 deletion testdata/dnn/onnx/generate_onnx_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -3042,7 +3042,7 @@ def make_node_with_constant(op_type, constant_value, inputs=None, outputs=None,
X = onnx.helper.make_tensor_value_info('input', onnx.TensorProto.FLOAT, input.shape)
Y = onnx.helper.make_tensor_value_info('output', onnx.TensorProto.FLOAT, output.shape)

shape = np.array([1, 1, -1], dtype=np.int32)
shape = np.array([1, 1, 1], dtype=np.int64)
shape = onnx.numpy_helper.from_array(shape, name='shape')
expand = onnx.helper.make_node("Expand", inputs=['embedding', 'shape'], outputs=['expand'])

Expand Down
Binary file modified testdata/dnn/onnx/models/clip-vit-base-head.onnx
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.

0 comments on commit 6608980

Please sign in to comment.