From d894b78c9554071331eb4051a539b1fa4c40028b Mon Sep 17 00:00:00 2001 From: Valery Chernov Date: Sun, 29 Jan 2023 17:51:02 +0300 Subject: [PATCH] graph fix --- tests/python/frontend/onnx/test_forward.py | 18 ++++++++++++++---- 1 file changed, 14 insertions(+), 4 deletions(-) diff --git a/tests/python/frontend/onnx/test_forward.py b/tests/python/frontend/onnx/test_forward.py index af15c879d70a3..63f13a8a46fa2 100644 --- a/tests/python/frontend/onnx/test_forward.py +++ b/tests/python/frontend/onnx/test_forward.py @@ -7766,7 +7766,7 @@ def verify_sequence_ops(tensor_shape, num_tensors, axis=0, position=0, new_axis= ) # Test sequence length - split_node = helper.make_node( + length_node = helper.make_node( "SequenceLength", inputs=["concat_sequence"], outputs=["output_2"] ) @@ -7787,10 +7787,20 @@ def verify_sequence_ops(tensor_shape, num_tensors, axis=0, position=0, new_axis= output_shape[axis] = num_tensors + 1 else: output_shape[axis] = (num_tensors + 1) * output_shape[axis] - graph_outputs = [helper.make_tensor_value_info("output", TensorProto.FLOAT, output_shape), - helper.make_tensor_value_info("output_2", TensorProto.INT, ())] + graph_outputs = [ + helper.make_tensor_value_info("output", TensorProto.FLOAT, output_shape), + helper.make_tensor_value_info("output_2", TensorProto.INT64, []), + ] - graph_nodes = [position_node, construct_node, insert_node, concat_node, split_node, at_node] + graph_nodes = [ + position_node, + construct_node, + insert_node, + concat_node, + split_node, + at_node, + length_node, + ] graph = helper.make_graph( graph_nodes,