Skip to content

GitLab

Commit 53349569 authored by Khalique's avatar Khalique
Browse files

return tuple for gen onnx script

parent 6b36c82e
Show whitespace changes
Inline Side-by-side
Showing with 108 additions and 489 deletions
test/onnx/gen_onnx.py
View file @ 53349569
...@@ -6,7 +6,23 @@ from onnx import AttributeProto, TensorProto, GraphProto ...@@ -6,7 +6,23 @@ from onnx import AttributeProto, TensorProto, GraphProto
def onnx_test(op_test): def onnx_test(op_test):
def run_test(): def run_test():
model_def = helper.make_model(op_test(), producer_name=op_test.__name__) op_info = op_test()
if len(op_info) > 3:
graph_def = helper.make_graph(
op_info[0],
op_test.__name__,
op_info[1],
op_info[2],
initializer=op_info[3]
)
else:
graph_def = helper.make_graph(
op_info[0],
op_test.__name__,
op_info[1],
op_info[2]
)
model_def = helper.make_model(graph_def, producer_name=op_test.__name__)
onnx.save(model_def, '{}.onnx'.format(op_test.__name__)) onnx.save(model_def, '{}.onnx'.format(op_test.__name__))
return run_test return run_test
...@@ -21,12 +37,7 @@ def acos_test(): ...@@ -21,12 +37,7 @@ def acos_test():
outputs=['y'], outputs=['y'],
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test_acos',
[x],
[y],
)
@onnx_test @onnx_test
def add_bcast_test(): def add_bcast_test():
...@@ -42,12 +53,7 @@ def add_bcast_test(): ...@@ -42,12 +53,7 @@ def add_bcast_test():
outputs=['2'] outputs=['2']
) )
return helper.make_graph( return ([node], [x,y], [z])
[node],
'test-add_bcast',
[x,y],
[z]
)
@onnx_test @onnx_test
def add_fp16_test(): def add_fp16_test():
...@@ -61,13 +67,12 @@ def add_fp16_test(): ...@@ -61,13 +67,12 @@ def add_fp16_test():
outputs=['2'], outputs=['2'],
) )
return helper.make_graph( return (
[node], [node],
'test-add-fp16',
[x,y], [x,y],
[z], [z],
# '0' -> 1.5, '1' -> 2.5 # '0' -> 1.5, '1' -> 2.5
initializer=[onnx.helper.make_tensor('0', TensorProto.FLOAT16, [1], [15872]), [onnx.helper.make_tensor('0', TensorProto.FLOAT16, [1], [15872]),
onnx.helper.make_tensor('1', TensorProto.FLOAT16, [1], [16640])] onnx.helper.make_tensor('1', TensorProto.FLOAT16, [1], [16640])]
) )
...@@ -86,12 +91,11 @@ def add_scalar_test(): ...@@ -86,12 +91,11 @@ def add_scalar_test():
outputs=['2'] outputs=['2']
) )
return helper.make_graph( return (
[node], [node],
'test-add-scalar',
[x,y], [x,y],
[z], [z],
initializer=[helper.make_tensor('1', TensorProto.FLOAT, [], [1])] [helper.make_tensor('1', TensorProto.FLOAT, [], [1])]
) )
@onnx_test @onnx_test
...@@ -107,13 +111,7 @@ def argmax_test(): ...@@ -107,13 +111,7 @@ def argmax_test():
keepdims = 0 keepdims = 0
) )
return ([node], [x], [y])
return helper.make_graph(
[node],
'test_argmax',
[x],
[y],
)
@onnx_test @onnx_test
def argmin_test(): def argmin_test():
...@@ -128,12 +126,7 @@ def argmin_test(): ...@@ -128,12 +126,7 @@ def argmin_test():
keepdims = 0 keepdims = 0
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test_argmin',
[x],
[y],
)
@onnx_test @onnx_test
def asin_test(): def asin_test():
...@@ -146,12 +139,8 @@ def asin_test(): ...@@ -146,12 +139,8 @@ def asin_test():
outputs=['y'], outputs=['y'],
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test_asin',
[x],
[y],
)
@onnx_test @onnx_test
def atan_test(): def atan_test():
...@@ -164,12 +153,7 @@ def atan_test(): ...@@ -164,12 +153,7 @@ def atan_test():
outputs=['y'], outputs=['y'],
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test_atan',
[x],
[y],
)
@onnx_test @onnx_test
def cast_test(): def cast_test():
...@@ -183,12 +167,7 @@ def cast_test(): ...@@ -183,12 +167,7 @@ def cast_test():
to = 1 to = 1
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test_cast',
[x],
[y],
)
@onnx_test @onnx_test
def clip_test(): def clip_test():
...@@ -204,12 +183,7 @@ def clip_test(): ...@@ -204,12 +183,7 @@ def clip_test():
min=0.0 min=0.0
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test-model',
[x],
[y]
)
@onnx_test @onnx_test
def concat_test(): def concat_test():
...@@ -224,12 +198,7 @@ def concat_test(): ...@@ -224,12 +198,7 @@ def concat_test():
outputs=['2'], outputs=['2'],
) )
return helper.make_graph( return ([node], [x,y], [z])
[node],
'test-concat',
[x,y],
[z]
)
@onnx_test @onnx_test
def constant_test(): def constant_test():
...@@ -248,12 +217,7 @@ def constant_test(): ...@@ -248,12 +217,7 @@ def constant_test():
), ),
) )
return helper.make_graph( return ([node], [], [y])
[node],
'test-constant',
[],
[y]
)
@onnx_test @onnx_test
def constant_fill_test(): def constant_fill_test():
...@@ -269,12 +233,7 @@ def constant_fill_test(): ...@@ -269,12 +233,7 @@ def constant_fill_test():
input_as_shape = 0, input_as_shape = 0,
) )
return helper.make_graph( return ([node], [], [value])
[node],
'constant_fill',
[],
[value],
)
@onnx_test @onnx_test
def constant_fill_input_as_shape_test(): def constant_fill_input_as_shape_test():
...@@ -305,12 +264,7 @@ def constant_fill_input_as_shape_test(): ...@@ -305,12 +264,7 @@ def constant_fill_input_as_shape_test():
input_as_shape = 1, input_as_shape = 1,
) )
return helper.make_graph( return ([const_shape_node, node], [], [value])
[const_shape_node, node],
'constant_fill',
[],
[value],
)
@onnx_test @onnx_test
def constant_scalar_test(): def constant_scalar_test():
...@@ -329,12 +283,7 @@ def constant_scalar_test(): ...@@ -329,12 +283,7 @@ def constant_scalar_test():
), ),
) )
return helper.make_graph( return ([node], [], [y])
[node],
'test-constant',
[],
[y]
)
@onnx_test @onnx_test
def const_of_shape_empty_input_test(): def const_of_shape_empty_input_test():
...@@ -365,12 +314,7 @@ def const_of_shape_empty_input_test(): ...@@ -365,12 +314,7 @@ def const_of_shape_empty_input_test():
value = tensor_val, value = tensor_val,
) )
return helper.make_graph( return ([shape_const, node], [], [y])
[shape_const, node],
'constant_of_shape',
[],
[y],
)
@onnx_test @onnx_test
def const_of_shape_float_test(): def const_of_shape_float_test():
...@@ -401,12 +345,7 @@ def const_of_shape_float_test(): ...@@ -401,12 +345,7 @@ def const_of_shape_float_test():
value = tensor_val value = tensor_val
) )
return helper.make_graph( return ([shape_const, node], [], [y])
[shape_const, node],
'constant_of_shape',
[],
[y],
)
@onnx_test @onnx_test
def const_of_shape_int64_test(): def const_of_shape_int64_test():
...@@ -436,12 +375,7 @@ def const_of_shape_int64_test(): ...@@ -436,12 +375,7 @@ def const_of_shape_int64_test():
value = tensor_val value = tensor_val
) )
return helper.make_graph( return ([shape_const, node], [], [y])
[shape_const, node],
'constant_of_shape',
[],
[y],
)
@onnx_test @onnx_test
def const_of_shape_no_value_attr_test(): def const_of_shape_no_value_attr_test():
...@@ -466,12 +400,7 @@ def const_of_shape_no_value_attr_test(): ...@@ -466,12 +400,7 @@ def const_of_shape_no_value_attr_test():
outputs=['y'], outputs=['y'],
) )
return helper.make_graph( return ([shape_const, node], [], [y])
[shape_const, node],
'constant_of_shape',
[],
[y],
)
@onnx_test @onnx_test
def conv_autopad_fail_test(): def conv_autopad_fail_test():
...@@ -489,12 +418,7 @@ def conv_autopad_fail_test(): ...@@ -489,12 +418,7 @@ def conv_autopad_fail_test():
pads = [0,0,1,1,0,0,1,1] pads = [0,0,1,1,0,0,1,1]
) )
return helper.make_graph( return ([node], [x,y], [out])
[node],
'test_conv',
[x, y],
[out],
)
@onnx_test @onnx_test
def conv_bias_test(): def conv_bias_test():
...@@ -511,12 +435,7 @@ def conv_bias_test(): ...@@ -511,12 +435,7 @@ def conv_bias_test():
strides = [1, 1] strides = [1, 1]
) )
return helper.make_graph( return ([node], [x,y,z], [out])
[node],
'test_conv',
[x, y, z],
[out],
)
@onnx_test @onnx_test
def conv_bn_relu_maxpool_test(): def conv_bn_relu_maxpool_test():
...@@ -560,11 +479,10 @@ def conv_bn_relu_maxpool_test(): ...@@ -560,11 +479,10 @@ def conv_bn_relu_maxpool_test():
kernel_shape=[2,2] kernel_shape=[2,2]
) )
return helper.make_graph( return (
[node0, node1, node2, node3], [node0, node1, node2, node3],
'test_conv_bn_relu',
[x, y, z, m, n, k, l], [x, y, z, m, n, k, l],
[out], [out]
) )
@onnx_test @onnx_test
...@@ -598,11 +516,10 @@ def conv_relu_maxpool_test(): ...@@ -598,11 +516,10 @@ def conv_relu_maxpool_test():
kernel_shape=[2,2] kernel_shape=[2,2]
) )
return helper.make_graph( return (
[node1, node2, node3], [node1, node2, node3],
'test_conv_relu',
[x, y, z], [x, y, z],
[out], [out]
) )
@onnx_test @onnx_test
...@@ -662,11 +579,10 @@ def conv_relu_maxpool_x2_test(): ...@@ -662,11 +579,10 @@ def conv_relu_maxpool_x2_test():
kernel_shape=[2,2] kernel_shape=[2,2]
) )
return helper.make_graph( return (
[node1, node2, node3, node4, node5, node6], [node1, node2, node3, node4, node5, node6],
'test_conv_relu2',
[x, y, z, m, n], [x, y, z, m, n],
[out], [out]
) )
@onnx_test @onnx_test
...@@ -680,12 +596,7 @@ def cos_test(): ...@@ -680,12 +596,7 @@ def cos_test():
outputs=['y'], outputs=['y'],
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test_cos',
[x],
[y],
)
@onnx_test @onnx_test
def cosh_test(): def cosh_test():
...@@ -698,12 +609,7 @@ def cosh_test(): ...@@ -698,12 +609,7 @@ def cosh_test():
outputs=['y'], outputs=['y'],
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test_cosh',
[x],
[y],
)
@onnx_test @onnx_test
def dropout_test(): def dropout_test():
...@@ -716,12 +622,7 @@ def dropout_test(): ...@@ -716,12 +622,7 @@ def dropout_test():
outputs=['1'], outputs=['1'],
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test-dropout',
[x],
[y]
)
@onnx_test @onnx_test
def elu_test(): def elu_test():
...@@ -735,12 +636,7 @@ def elu_test(): ...@@ -735,12 +636,7 @@ def elu_test():
alpha=0.01 alpha=0.01
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test-model',
[x],
[y]
)
@onnx_test @onnx_test
def erf_test(): def erf_test():
...@@ -753,12 +649,7 @@ def erf_test(): ...@@ -753,12 +649,7 @@ def erf_test():
outputs=['y'], outputs=['y'],
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test_erf',
[x],
[y],
)
@onnx_test @onnx_test
def exp_test(): def exp_test():
...@@ -771,12 +662,7 @@ def exp_test(): ...@@ -771,12 +662,7 @@ def exp_test():
outputs=['y'], outputs=['y'],
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test_exp',
[x],
[y],
)
@onnx_test @onnx_test
def expand_test(): def expand_test():
...@@ -802,12 +688,7 @@ def expand_test(): ...@@ -802,12 +688,7 @@ def expand_test():
outputs=['y'] outputs=['y']
) )
return helper.make_graph( return ([shape_const,node], [x], [y])
[shape_const, node],
'expand',
[x],
[y],
)
@onnx_test @onnx_test
def flatten_test(): def flatten_test():
...@@ -828,12 +709,7 @@ def flatten_test(): ...@@ -828,12 +709,7 @@ def flatten_test():
outputs=['3'] outputs=['3']
) )
return helper.make_graph( return ([node,node2], [x], [y,y2])
[node,node2],
'test-flatten',
[x],
[y,y2]
)
@onnx_test @onnx_test
def gather_test(): def gather_test():
...@@ -848,12 +724,7 @@ def gather_test(): ...@@ -848,12 +724,7 @@ def gather_test():
axis=1, axis=1,
) )
return helper.make_graph( return ([node], [x,i], [y])
[node],
'test_gather',
[x, i],
[y],
)
@onnx_test @onnx_test
def gemm_test(): def gemm_test():
...@@ -872,12 +743,7 @@ def gemm_test(): ...@@ -872,12 +743,7 @@ def gemm_test():
transB=1 transB=1
) )
return helper.make_graph( return ([node], [x,y,z], [a])
[node],
'test-gemm',
[x, y, z],
[a]
)
@onnx_test @onnx_test
def gemm_ex_test(): def gemm_ex_test():
...@@ -895,12 +761,7 @@ def gemm_ex_test(): ...@@ -895,12 +761,7 @@ def gemm_ex_test():
transA = 1 transA = 1
) )
return helper.make_graph( return ([node], [m1,m2,m3], [y])
[node],
'test_gemm_ex',
[m1, m2, m3],
[y],
)
@onnx_test @onnx_test
def gemm_ex_brcst_test(): def gemm_ex_brcst_test():
...@@ -918,12 +779,7 @@ def gemm_ex_brcst_test(): ...@@ -918,12 +779,7 @@ def gemm_ex_brcst_test():
transA = 1 transA = 1
) )
return helper.make_graph( return ([node], [m1,m2,m3], [y])
[node],
'test_gemm_ex',
[m1, m2, m3],
[y],
)
@onnx_test @onnx_test
def globalavgpool_test(): def globalavgpool_test():
...@@ -936,12 +792,7 @@ def globalavgpool_test(): ...@@ -936,12 +792,7 @@ def globalavgpool_test():
outputs=['1'], outputs=['1'],
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test-globalavgpool',
[x],
[y]
)
@onnx_test @onnx_test
def globalmaxpool_test(): def globalmaxpool_test():
...@@ -954,12 +805,7 @@ def globalmaxpool_test(): ...@@ -954,12 +805,7 @@ def globalmaxpool_test():
outputs=['1'], outputs=['1'],
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test-globalmaxpool',
[x],
[y]
)
@onnx_test @onnx_test
def group_conv_test(): def group_conv_test():
...@@ -974,12 +820,7 @@ def group_conv_test(): ...@@ -974,12 +820,7 @@ def group_conv_test():
outputs=['2'], outputs=['2'],
) )
return helper.make_graph( return ([node], [x,y], [z])
[node],
'test-group_conv',
[x,y],
[z]
)
@onnx_test @onnx_test
def imagescaler_test(): def imagescaler_test():
...@@ -994,12 +835,7 @@ def imagescaler_test(): ...@@ -994,12 +835,7 @@ def imagescaler_test():
scale=0.5 scale=0.5
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test-imagescaler',
[x],
[y]
)
@onnx_test @onnx_test
def implicit_add_bcast_test(): def implicit_add_bcast_test():
...@@ -1013,12 +849,7 @@ def implicit_add_bcast_test(): ...@@ -1013,12 +849,7 @@ def implicit_add_bcast_test():
outputs=['2'], outputs=['2'],
) )
return helper.make_graph( return ([node], [x,y], [z])
[node],
'test-multi_bcast',
[x,y],
[z]
)
@onnx_test @onnx_test
def implicit_pow_bcast_test(): def implicit_pow_bcast_test():
...@@ -1032,12 +863,7 @@ def implicit_pow_bcast_test(): ...@@ -1032,12 +863,7 @@ def implicit_pow_bcast_test():
outputs=['out'], outputs=['out'],
) )
return helper.make_graph( return ([node], [arg0,arg1], [arg_out])
[node],
'pow_test',
[arg0, arg1],
[arg_out],
)
@onnx_test @onnx_test
def implicit_sub_bcast_test(): def implicit_sub_bcast_test():
...@@ -1051,12 +877,7 @@ def implicit_sub_bcast_test(): ...@@ -1051,12 +877,7 @@ def implicit_sub_bcast_test():
outputs=['out'], outputs=['out'],
) )
return helper.make_graph( return ([node], [arg0,arg1], [arg_out])
[node],
'subtraction2',
[arg0, arg1],
[arg_out],
)
@onnx_test @onnx_test
def leaky_relu_test(): def leaky_relu_test():
...@@ -1070,12 +891,7 @@ def leaky_relu_test(): ...@@ -1070,12 +891,7 @@ def leaky_relu_test():
alpha=0.01 alpha=0.01
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test-model',
[x],
[y]
)
@onnx_test @onnx_test
def log_test(): def log_test():
...@@ -1088,12 +904,7 @@ def log_test(): ...@@ -1088,12 +904,7 @@ def log_test():
outputs=['y'], outputs=['y'],
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test_log',
[x],
[y],
)
@onnx_test @onnx_test
def logsoftmax_test(): def logsoftmax_test():
...@@ -1107,12 +918,7 @@ def logsoftmax_test(): ...@@ -1107,12 +918,7 @@ def logsoftmax_test():
axis = 1 axis = 1
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test_logsoftmax',
[x],
[y]
)
@onnx_test @onnx_test
def lrn_test(): def lrn_test():
...@@ -1129,12 +935,7 @@ def lrn_test(): ...@@ -1129,12 +935,7 @@ def lrn_test():
outputs=['1'] outputs=['1']
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test-lrn',
[x],
[y]
)
@onnx_test @onnx_test
def matmul_bmbm_test(): def matmul_bmbm_test():
...@@ -1148,12 +949,7 @@ def matmul_bmbm_test(): ...@@ -1148,12 +949,7 @@ def matmul_bmbm_test():
outputs=['y'], outputs=['y'],
) )
return helper.make_graph( return ([node], [m1,m2], [y])
[node],
'test_matmul',
[m1, m2],
[y],
)
@onnx_test @onnx_test
def matmul_bmv_test(): def matmul_bmv_test():
...@@ -1167,12 +963,7 @@ def matmul_bmv_test(): ...@@ -1167,12 +963,7 @@ def matmul_bmv_test():
outputs=['y'], outputs=['y'],
) )
return helper.make_graph( return ([node], [m1,m2], [y])
[node],
'test_matmul',
[m1, m2],
[y],
)
@onnx_test @onnx_test
def matmul_mv_test(): def matmul_mv_test():
...@@ -1186,12 +977,7 @@ def matmul_mv_test(): ...@@ -1186,12 +977,7 @@ def matmul_mv_test():
outputs=['y'], outputs=['y'],
) )
return helper.make_graph( return ([node], [m1,m2], [y])
[node],
'test_matmul',
[m1, m2],
[y],
)
@onnx_test @onnx_test
def matmul_vbm_test(): def matmul_vbm_test():
...@@ -1205,12 +991,7 @@ def matmul_vbm_test(): ...@@ -1205,12 +991,7 @@ def matmul_vbm_test():
outputs=['y'], outputs=['y'],
) )
return helper.make_graph( return ([node], [m1,m2], [y])
[node],
'test_matmul',
[m1, m2],
[y],
)
@onnx_test @onnx_test
def matmul_vm_test(): def matmul_vm_test():
...@@ -1224,12 +1005,7 @@ def matmul_vm_test(): ...@@ -1224,12 +1005,7 @@ def matmul_vm_test():
outputs=['y'], outputs=['y'],
) )
return helper.make_graph( return ([node], [m1,m2], [y])
[node],
'test_matmul',
[m1, m2],
[y],
)
@onnx_test @onnx_test
def matmul_vv_test(): def matmul_vv_test():
...@@ -1243,12 +1019,7 @@ def matmul_vv_test(): ...@@ -1243,12 +1019,7 @@ def matmul_vv_test():
outputs=['y'], outputs=['y'],
) )
return helper.make_graph( return ([node], [m1,m2], [y])
[node],
'test_matmul',
[m1, m2],
[y],
)
@onnx_test @onnx_test
def max_test(): def max_test():
...@@ -1263,12 +1034,7 @@ def max_test(): ...@@ -1263,12 +1034,7 @@ def max_test():
outputs=['3'], outputs=['3'],
) )
return helper.make_graph( return ([node], [a,b,c], [y])
[node],
'test-dropout',
[a, b, c],
[y]
)
@onnx_test @onnx_test
def min_test(): def min_test():
...@@ -1283,12 +1049,7 @@ def min_test(): ...@@ -1283,12 +1049,7 @@ def min_test():
outputs=['3'], outputs=['3'],
) )
return helper.make_graph( return ([node], [a,b,c], [y])
[node],
'test-dropout',
[a, b, c],
[y]
)
@onnx_test @onnx_test
def no_pad_test(): def no_pad_test():
...@@ -1302,13 +1063,7 @@ def no_pad_test(): ...@@ -1302,13 +1063,7 @@ def no_pad_test():
outputs=['1'] outputs=['1']
) )
return ([node], [x], [y])
return helper.make_graph(
[node],
'test-no-pad',
[x],
[y]
)
@onnx_test @onnx_test
def pad_test(): def pad_test():
...@@ -1322,13 +1077,7 @@ def pad_test(): ...@@ -1322,13 +1077,7 @@ def pad_test():
outputs=['1'] outputs=['1']
) )
return ([node], [x], [y])
return helper.make_graph(
[node],
'test-pad',
[x],
[y]
)
@onnx_test @onnx_test
def pow_test(): def pow_test():
...@@ -1343,12 +1092,7 @@ def pow_test(): ...@@ -1343,12 +1092,7 @@ def pow_test():
) )
return helper.make_graph( return ([node], [arg0, arg1], [arg_out])
[node],
'pow_test',
[arg0, arg1],
[arg_out],
)
@onnx_test @onnx_test
def reducemean_test(): def reducemean_test():
...@@ -1364,12 +1108,7 @@ def reducemean_test(): ...@@ -1364,12 +1108,7 @@ def reducemean_test():
keepdims = 0 keepdims = 0
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test_reducemean',
[x],
[y],
)
@onnx_test @onnx_test
def reducemean_keepdims_test(): def reducemean_keepdims_test():
...@@ -1385,12 +1124,7 @@ def reducemean_keepdims_test(): ...@@ -1385,12 +1124,7 @@ def reducemean_keepdims_test():
keepdims = 1 keepdims = 1
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test_reducemean',
[x],
[y],
)
@onnx_test @onnx_test
def reducesum_test(): def reducesum_test():
...@@ -1406,12 +1140,7 @@ def reducesum_test(): ...@@ -1406,12 +1140,7 @@ def reducesum_test():
keepdims = 0 keepdims = 0
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test_reducesum',
[x],
[y],
)
@onnx_test @onnx_test
def reducesum_multiaxis_test(): def reducesum_multiaxis_test():
...@@ -1427,12 +1156,7 @@ def reducesum_multiaxis_test(): ...@@ -1427,12 +1156,7 @@ def reducesum_multiaxis_test():
keepdims = 0 keepdims = 0
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test_reducesum',
[x],
[y],
)
@onnx_test @onnx_test
def reducesum_keepdims_test(): def reducesum_keepdims_test():
...@@ -1448,12 +1172,7 @@ def reducesum_keepdims_test(): ...@@ -1448,12 +1172,7 @@ def reducesum_keepdims_test():
keepdims = 1 keepdims = 1
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test_reducesum',
[x],
[y],
)
@onnx_test @onnx_test
def reshape_test(): def reshape_test():
...@@ -1476,12 +1195,11 @@ def reshape_test(): ...@@ -1476,12 +1195,11 @@ def reshape_test():
outputs=['3'] outputs=['3']
) )
return helper.make_graph( return (
[node,node2], [node,node2],
'test-reshape',
[x, x_shape], [x, x_shape],
[y,y2], [y,y2],
initializer=[helper.make_tensor('1', TensorProto.INT64, [2], [3, 8])] [helper.make_tensor('1', TensorProto.INT64, [2], [3, 8])]
) )
@onnx_test @onnx_test
...@@ -1504,12 +1222,7 @@ def reshape_non_standard_test(): ...@@ -1504,12 +1222,7 @@ def reshape_non_standard_test():
shape=[4, 3, 2] shape=[4, 3, 2]
) )
return helper.make_graph( return ([trans,res], [x], [y])
[trans, res],
'reshape-ns',
[x],
[y],
)
@onnx_test @onnx_test
def shape_test(): def shape_test():
...@@ -1522,12 +1235,7 @@ def shape_test(): ...@@ -1522,12 +1235,7 @@ def shape_test():
outputs=['y'], outputs=['y'],
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test_shape',
[x],
[y],
)
@onnx_test @onnx_test
def shape_gather_test(): def shape_gather_test():
...@@ -1563,12 +1271,7 @@ def shape_gather_test(): ...@@ -1563,12 +1271,7 @@ def shape_gather_test():
axis=0, axis=0,
) )
return helper.make_graph( return ([node_const,node_shape,node_gather], [x], [z])
[node_const, node_shape, node_gather],
'shape_gather',
[x],
[z],
)
@onnx_test @onnx_test
def sign_test(): def sign_test():
...@@ -1581,12 +1284,7 @@ def sign_test(): ...@@ -1581,12 +1284,7 @@ def sign_test():
outputs=['y'], outputs=['y'],
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test_sign',
[x],
[y],
)
@onnx_test @onnx_test
def sin_test(): def sin_test():
...@@ -1599,12 +1297,7 @@ def sin_test(): ...@@ -1599,12 +1297,7 @@ def sin_test():
outputs=['y'], outputs=['y'],
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test_sin',
[x],
[y],
)
@onnx_test @onnx_test
def sinh_test(): def sinh_test():
...@@ -1617,12 +1310,7 @@ def sinh_test(): ...@@ -1617,12 +1310,7 @@ def sinh_test():
outputs=['y'], outputs=['y'],
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test_sinh',
[x],
[y],
)
@onnx_test @onnx_test
def slice_test(): def slice_test():
...@@ -1638,12 +1326,7 @@ def slice_test(): ...@@ -1638,12 +1326,7 @@ def slice_test():
outputs=['1'] outputs=['1']
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test-slice',
[x],
[y]
)
@onnx_test @onnx_test
def softmax_test(): def softmax_test():
...@@ -1656,12 +1339,7 @@ def softmax_test(): ...@@ -1656,12 +1339,7 @@ def softmax_test():
outputs=['1'] outputs=['1']
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test-softmax',
[x],
[y]
)
@onnx_test @onnx_test
def sqrt_test(): def sqrt_test():
...@@ -1674,12 +1352,7 @@ def sqrt_test(): ...@@ -1674,12 +1352,7 @@ def sqrt_test():
outputs=['y'], outputs=['y'],
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test_sqrt',
[x],
[y],
)
@onnx_test @onnx_test
def squeeze_unsqueeze_test(): def squeeze_unsqueeze_test():
...@@ -1701,12 +1374,7 @@ def squeeze_unsqueeze_test(): ...@@ -1701,12 +1374,7 @@ def squeeze_unsqueeze_test():
outputs=['2'] outputs=['2']
) )
return helper.make_graph( return ([node,node2], [x], [z])
[node,node2],
'test-squeeze-unsqueeze',
[x],
[z]
)
@onnx_test @onnx_test
def sub_bcast_test(): def sub_bcast_test():
...@@ -1722,13 +1390,7 @@ def sub_bcast_test(): ...@@ -1722,13 +1390,7 @@ def sub_bcast_test():
axis = 1, axis = 1,
) )
return ([node], [arg0,arg1], [arg_out])
return helper.make_graph(
[node],
'subtraction2',
[arg0, arg1],
[arg_out],
)
@onnx_test @onnx_test
def sub_scalar_test(): def sub_scalar_test():
...@@ -1757,12 +1419,7 @@ def sub_scalar_test(): ...@@ -1757,12 +1419,7 @@ def sub_scalar_test():
outputs=['out'], outputs=['out'],
) )
return helper.make_graph( return ([arg_const,node], [arg_node], [arg_out])
[arg_const, node],
'subtraction1',
[arg_node],
[arg_out],
)
@onnx_test @onnx_test
def sum_test(): def sum_test():
...@@ -1778,12 +1435,7 @@ def sum_test(): ...@@ -1778,12 +1435,7 @@ def sum_test():
outputs=['3'], outputs=['3'],
) )
return helper.make_graph( return ([node], [a,b,c], [y])
[node],
'test-sum',
[a, b, c],
[y]
)
@onnx_test @onnx_test
def sum_test(): def sum_test():
...@@ -1798,12 +1450,7 @@ def sum_test(): ...@@ -1798,12 +1450,7 @@ def sum_test():
outputs=['3'], outputs=['3'],
) )
return helper.make_graph( return ([node], [a,b,c], [y])
[node],
'test-sum',
[a, b, c],
[y]
)
@onnx_test @onnx_test
def tan_test(): def tan_test():
...@@ -1816,12 +1463,7 @@ def tan_test(): ...@@ -1816,12 +1463,7 @@ def tan_test():
outputs=['y'], outputs=['y'],
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test_tan',
[x],
[y],
)
@onnx_test @onnx_test
def tanh_test(): def tanh_test():
...@@ -1834,12 +1476,7 @@ def tanh_test(): ...@@ -1834,12 +1476,7 @@ def tanh_test():
outputs=['y'], outputs=['y'],
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test_tanh',
[x],
[y],
)
@onnx_test @onnx_test
def transpose_test(): def transpose_test():
...@@ -1853,12 +1490,7 @@ def transpose_test(): ...@@ -1853,12 +1490,7 @@ def transpose_test():
outputs=['1'], outputs=['1'],
) )
return helper.make_graph( return ([node], [x], [y])
[node],
'test-transpose',
[x],
[y]
)
@onnx_test @onnx_test
def transpose_gather_test(): def transpose_gather_test():
...@@ -1888,12 +1520,7 @@ def transpose_gather_test(): ...@@ -1888,12 +1520,7 @@ def transpose_gather_test():
) )
return helper.make_graph( return ([td, ti, node], [x, i], [y])
[td, ti, node],
'test_gather',
[x, i],
[y],
)
@onnx_test @onnx_test
def unknown_test(): def unknown_test():
...@@ -1914,12 +1541,4 @@ def unknown_test(): ...@@ -1914,12 +1541,4 @@ def unknown_test():
outputs=['3'] outputs=['3']
) )
return helper.make_graph( return ([node,node2], [x,y], [a])
[node,node2], \ No newline at end of file
'test-unknown',
[x,y],
[a]
)
model_def = helper.make_model(graph_def, producer_name='unknown-example')
onnx.save(model_def, 'unknown_test.onnx')
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment