"runner/vscode:/vscode.git/clone" did not exist on "7d25b9e194f106e9c2a5289dfde40077c0838b7d"
Commit 571fc2cd authored by Paul's avatar Paul
Browse files

Remove broadcast check in generic op

parent da2f772a
......@@ -163,19 +163,6 @@ struct onnx_parser
void add_generic_op(std::string name, T x)
{
ops.emplace(name, [this, x](attribute_map attributes, std::vector<instruction_ref> args) {
if(args.size() == 2 and contains(attributes, "broadcast"))
{
uint64_t broadcasted = parse_value(attributes.at("broadcast")).at<uint64_t>();
if(broadcasted != 0)
{
uint64_t axis = (contains(attributes, "axis"))
? parse_value(attributes.at("axis")).at<uint64_t>()
: 0;
auto l =
prog.add_instruction(op::broadcast{axis, args[0]->get_shape()}, args[1]);
return prog.add_instruction(x, args[0], l);
}
}
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