Skip to content

GitLab

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

Fix format

parent 2ae4c715
Show 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,
...@@ -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'],
...@@ -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,
...@@ -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