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

Fix issue with if then/else cases for trailing one

Inverted logic when I wrote generated statements for these. Regenerated files
parent 3e067a8a
...@@ -2261,7 +2261,7 @@ def if_else_empty_shape_test(): ...@@ -2261,7 +2261,7 @@ def if_else_empty_shape_test():
@onnx_test @onnx_test
def if_else_trailing_one_shape_test(): def if_then_trailing_one_shape_test():
x = onnx.helper.make_tensor_value_info('x', onnx.TensorProto.FLOAT, [2]) x = onnx.helper.make_tensor_value_info('x', onnx.TensorProto.FLOAT, [2])
y = onnx.helper.make_tensor_value_info('y', onnx.TensorProto.FLOAT, [2, 1]) y = onnx.helper.make_tensor_value_info('y', onnx.TensorProto.FLOAT, [2, 1])
...@@ -2298,7 +2298,7 @@ def if_else_trailing_one_shape_test(): ...@@ -2298,7 +2298,7 @@ def if_else_trailing_one_shape_test():
else_body = onnx.helper.make_graph([else_mul_node], 'else_body', [], else_body = onnx.helper.make_graph([else_mul_node], 'else_body', [],
[else_out]) [else_out])
cond = np.array([0]).astype(np.bool) cond = np.array([1]).astype(np.bool)
cond_tensor = helper.make_tensor(name="cond", cond_tensor = helper.make_tensor(name="cond",
data_type=TensorProto.BOOL, data_type=TensorProto.BOOL,
dims=cond.shape, dims=cond.shape,
...@@ -2570,7 +2570,7 @@ def if_pl_test(): ...@@ -2570,7 +2570,7 @@ def if_pl_test():
@onnx_test @onnx_test
def if_then_trailing_one_shape_test(): def if_else_trailing_one_shape_test():
x = onnx.helper.make_tensor_value_info('x', onnx.TensorProto.FLOAT, [2, 1]) x = onnx.helper.make_tensor_value_info('x', onnx.TensorProto.FLOAT, [2, 1])
y = onnx.helper.make_tensor_value_info('y', onnx.TensorProto.FLOAT, [2]) y = onnx.helper.make_tensor_value_info('y', onnx.TensorProto.FLOAT, [2])
...@@ -2607,7 +2607,7 @@ def if_then_trailing_one_shape_test(): ...@@ -2607,7 +2607,7 @@ def if_then_trailing_one_shape_test():
else_body = onnx.helper.make_graph([else_mul_node], 'else_body', [], else_body = onnx.helper.make_graph([else_mul_node], 'else_body', [],
[else_out]) [else_out])
cond = np.array([1]).astype(np.bool) cond = np.array([0]).astype(np.bool)
cond_tensor = helper.make_tensor(name="cond", cond_tensor = helper.make_tensor(name="cond",
data_type=TensorProto.BOOL, data_type=TensorProto.BOOL,
dims=cond.shape, dims=cond.shape,
......
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