Skip to content

GitLab

Commit d8952ebf authored by Ted Themistokleous's avatar Ted Themistokleous
Browse files

Fix format

parent 2ae4c715
Hide whitespace changes
Inline Side-by-side
Showing with 17 additions and 15 deletions
test/onnx/gen_onnx.py
View file @ d8952ebf
...@@ -2870,8 +2870,8 @@ def if_else_empty_constant_multi_output_test(): ...@@ -2870,8 +2870,8 @@ def if_else_empty_constant_multi_output_test():
else_out = helper.make_tensor_value_info('shape_const', TensorProto.INT64, else_out = helper.make_tensor_value_info('shape_const', TensorProto.INT64,
[]) [])
else_out2 = helper.make_tensor_value_info('shape_const2', TensorProto.INT64, else_out2 = helper.make_tensor_value_info('shape_const2',
[]) TensorProto.INT64, [])
then_out = onnx.helper.make_tensor_value_info('then_out', then_out = onnx.helper.make_tensor_value_info('then_out',
onnx.TensorProto.INT64, onnx.TensorProto.INT64,
...@@ -2889,13 +2889,13 @@ def if_else_empty_constant_multi_output_test(): ...@@ -2889,13 +2889,13 @@ def if_else_empty_constant_multi_output_test():
inputs=['y', 'yt'], inputs=['y', 'yt'],
outputs=['then_out']) outputs=['then_out'])
then_sub_node = onnx.helper.make_node('Sub', then_sub_node = onnx.helper.make_node('Sub',
inputs=['y', 'yt'], inputs=['y', 'yt'],
outputs=['then_out2']) outputs=['then_out2'])
else_body = onnx.helper.make_graph([shape_const, shape_const2], else_body = onnx.helper.make_graph([shape_const, shape_const2],
'else_body', [], [else_out, else_out2]) 'else_body', [], [else_out, else_out2])
then_body = onnx.helper.make_graph([then_mul_node, then_sub_node], then_body = onnx.helper.make_graph([then_mul_node, then_sub_node],
'then_body', [], [then_out, then_out2]) 'then_body', [], [then_out, then_out2])
cond = np.array([0]).astype(np.bool) cond = np.array([0]).astype(np.bool)
...@@ -2904,7 +2904,8 @@ def if_else_empty_constant_multi_output_test(): ...@@ -2904,7 +2904,8 @@ def if_else_empty_constant_multi_output_test():
dims=cond.shape, dims=cond.shape,
vals=cond.astype(bool)) vals=cond.astype(bool))
res = onnx.helper.make_tensor_value_info('res', TensorProto.INT64, [2, 1]) res = onnx.helper.make_tensor_value_info('res', TensorProto.INT64, [2, 1])
res2 = onnx.helper.make_tensor_value_info('res2', TensorProto.INT64, [2, 1]) res2 = onnx.helper.make_tensor_value_info('res2', TensorProto.INT64,
[2, 1])
node = onnx.helper.make_node('If', node = onnx.helper.make_node('If',
inputs=['cond'], inputs=['cond'],
...@@ -2984,7 +2985,7 @@ def if_then_empty_constant_multi_output_test(): ...@@ -2984,7 +2985,7 @@ def if_then_empty_constant_multi_output_test():
outputs=['shape_const'], outputs=['shape_const'],
value=empty_ts, value=empty_ts,
) )
shape_const2 = helper.make_node( shape_const2 = helper.make_node(
'Constant', 'Constant',
inputs=[], inputs=[],
...@@ -2996,8 +2997,8 @@ def if_then_empty_constant_multi_output_test(): ...@@ -2996,8 +2997,8 @@ def if_then_empty_constant_multi_output_test():
then_out = helper.make_tensor_value_info('shape_const', TensorProto.INT64, then_out = helper.make_tensor_value_info('shape_const', TensorProto.INT64,
[]) [])
then_out2 = helper.make_tensor_value_info('shape_const2', TensorProto.INT64, then_out2 = helper.make_tensor_value_info('shape_const2',
[]) TensorProto.INT64, [])
else_out = onnx.helper.make_tensor_value_info('else_out', else_out = onnx.helper.make_tensor_value_info('else_out',
onnx.TensorProto.INT64, onnx.TensorProto.INT64,
...@@ -3016,13 +3017,13 @@ def if_then_empty_constant_multi_output_test(): ...@@ -3016,13 +3017,13 @@ def if_then_empty_constant_multi_output_test():
inputs=['y', 'yt'], inputs=['y', 'yt'],
outputs=['else_out']) outputs=['else_out'])
else_sub_node = onnx.helper.make_node('Sub', else_sub_node = onnx.helper.make_node('Sub',
inputs=['y', 'yt'], inputs=['y', 'yt'],
outputs=['else_out2']) outputs=['else_out2'])
else_body = onnx.helper.make_graph([else_mul_node, else_sub_node], else_body = onnx.helper.make_graph([else_mul_node, else_sub_node],
'else_body', [], [else_out, else_out2]) 'else_body', [], [else_out, else_out2])
then_body = onnx.helper.make_graph([shape_const, shape_const2], then_body = onnx.helper.make_graph([shape_const, shape_const2],
'then_body', [], [then_out, then_out2]) 'then_body', [], [then_out, then_out2])
cond = np.array([1]).astype(np.bool) cond = np.array([1]).astype(np.bool)
...@@ -3031,7 +3032,8 @@ def if_then_empty_constant_multi_output_test(): ...@@ -3031,7 +3032,8 @@ def if_then_empty_constant_multi_output_test():
dims=cond.shape, dims=cond.shape,
vals=cond.astype(bool)) vals=cond.astype(bool))
res = onnx.helper.make_tensor_value_info('res', TensorProto.INT64, [2, 1]) res = onnx.helper.make_tensor_value_info('res', TensorProto.INT64, [2, 1])
res2 = onnx.helper.make_tensor_value_info('res2', TensorProto.INT64, [2, 1]) res2 = onnx.helper.make_tensor_value_info('res2', TensorProto.INT64,
[2, 1])
node = onnx.helper.make_node('If', node = onnx.helper.make_node('If',
inputs=['cond'], inputs=['cond'],
......
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