Commit b9ec1d6d authored by Paul's avatar Paul
Browse files

Format

parent 8429ff39
...@@ -341,14 +341,15 @@ struct find_inner_broadcast ...@@ -341,14 +341,15 @@ struct find_inner_broadcast
std::back_inserter(inputs), std::back_inserter(inputs),
[](auto i) { return i->inputs().front(); }); [](auto i) { return i->inputs().front(); });
if(std::any_of(inputs.begin(), inputs.end(), [&](auto i) { if(std::any_of(inputs.begin(), inputs.end(), [&](auto i) {
return i->get_shape() != inputs.front()->get_shape() and i->get_shape().elements() != 1; return i->get_shape() != inputs.front()->get_shape() and
i->get_shape().elements() != 1;
})) }))
return; return;
auto b_it = std::find_if(broadcasts.begin(), broadcasts.end(), [&](auto i) { auto b_it = std::find_if(broadcasts.begin(), broadcasts.end(), [&](auto i) {
return not i->get_shape().scalar(); return not i->get_shape().scalar();
}); });
if (b_it == broadcasts.end()) if(b_it == broadcasts.end())
b_it = broadcasts.begin(); b_it = broadcasts.begin();
auto op = insert_common_op(m, ins, ins->get_operator(), inputs); auto op = insert_common_op(m, ins, ins->get_operator(), inputs);
m.replace_instruction(ins, (*b_it)->get_operator(), op); m.replace_instruction(ins, (*b_it)->get_operator(), op);
......
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