Skip to content

GitLab

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

Only check types are equal, leave size checks till later.

parent 8b08a86f
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
src/onnx/parse_if.cpp
View file @ 69977052
...@@ -70,8 +70,7 @@ struct parse_if : op_parser<parse_if> ...@@ -70,8 +70,7 @@ struct parse_if : op_parser<parse_if>
// Must have the same type for both if/else blocks by onnx spec // Must have the same type for both if/else blocks by onnx spec
// Add exception for empty constant scalars // Add exception for empty constant scalars
if(then_out_shapes.at(0).type() != else_out_shapes.at(0).type() && if(then_out_shapes.at(0).type() != else_out_shapes.at(0).type())
(then_out_shapes.at(0).elements() > 0) && (else_out_shapes.at(0).elements() > 0))
{ {
MIGRAPHX_THROW("PARSE_IF: " + info.name + MIGRAPHX_THROW("PARSE_IF: " + info.name +
" then and else sub_grahps must have same output type! " + " then and else sub_grahps must have same output type! " +
......
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