Commit 80203608 authored by Paul's avatar Paul
Browse files

Merge branch 'multi_broadcast-fix'

parents 3ef07c55 c74767b1
......@@ -113,7 +113,7 @@ struct onnx_parser
}
return prog.add_instruction(x, args);
}
else
else if(args[0]->get_shape() != args[1]->get_shape())
{
// Example:
// s0 = (3,2,4,5) and s1 = (2,1,1)
......@@ -149,6 +149,10 @@ struct onnx_parser
auto l1 = prog.add_instruction(op::multibroadcast{output_lens}, args[1]);
return prog.add_instruction(x, l0, l1);
}
else
{
return prog.add_instruction(x, args);
}
});
}
......
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