Commit d771f6b5 authored by Scott Thornton's avatar Scott Thornton
Browse files

Formatting

parent 1f304aed
...@@ -777,19 +777,19 @@ struct multibroadcast ...@@ -777,19 +777,19 @@ struct multibroadcast
auto t = inputs.at(0).type(); auto t = inputs.at(0).type();
auto input = inputs.at(0); auto input = inputs.at(0);
if (input.lens().size() <= 0) if(input.lens().size() <= 0)
MIGRAPH_THROW("inputs dimensions should be > 0"); MIGRAPH_THROW("inputs dimensions should be > 0");
if (input.lens().size() > output_lens.size()) if(input.lens().size() > output_lens.size())
MIGRAPH_THROW("inputs dimensions should <= output size"); MIGRAPH_THROW("inputs dimensions should <= output size");
std::vector<size_t> bcast_strides(output_lens.size(), 0); std::vector<size_t> bcast_strides(output_lens.size(), 0);
auto offset = output_lens.size()-input.lens().size(); auto offset = output_lens.size() - input.lens().size();
for (int i = input.lens().size()-1; i >= 0; i--) for(int i = input.lens().size() - 1; i >= 0; i--)
{ {
if (output_lens[i+offset] == input.lens()[i]) if(output_lens[i + offset] == input.lens()[i])
{ {
bcast_strides[i+offset] = input.strides()[i]; bcast_strides[i + offset] = input.strides()[i];
} }
} }
return {t, output_lens, bcast_strides}; return {t, output_lens, bcast_strides};
......
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