Commit 483c3e56 authored by Ted Themistokleous's avatar Ted Themistokleous Committed by Ted Themistokleous
Browse files

Handle static shapes only for conversion

Need to handle static shapes explicitly since onnx says we should be able to just output a compatible output type.
parent 83215963
...@@ -77,7 +77,7 @@ struct parse_if : op_parser<parse_if> ...@@ -77,7 +77,7 @@ struct parse_if : op_parser<parse_if>
else_out_shapes.at(0).type_string()); else_out_shapes.at(0).type_string());
} }
if(not then_out_shapes.at(0).scalar() && not else_out_shapes.at(0).scalar()) if(not then_out_shapes.at(0).dynamic() && not else_out_shapes.at(0).dynamic())
{ {
// First dimension must agree // First dimension must agree
if(then_out_shapes.at(0).lens().at(0) != else_out_shapes.at(0).lens().at(0)) if(then_out_shapes.at(0).lens().at(0) != else_out_shapes.at(0).lens().at(0))
......
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