Commit 868230f5 authored by Paul's avatar Paul
Browse files

Format

parent 14ec0ee9
...@@ -85,7 +85,9 @@ preload preload::broadcasts(std::size_t axis, const std::vector<shape>& inputs) ...@@ -85,7 +85,9 @@ preload preload::broadcasts(std::size_t axis, const std::vector<shape>& inputs)
std::vector<bool> result(inputs.size()); std::vector<bool> result(inputs.size());
std::vector<std::size_t> preloaded; std::vector<std::size_t> preloaded;
auto idxs = range(inputs.size()); auto idxs = range(inputs.size());
std::copy_if(idxs.begin(), idxs.end(), std::back_inserter(preloaded), [&](auto i){ return inputs[i].strides()[axis] == 0; }); std::copy_if(idxs.begin(), idxs.end(), std::back_inserter(preloaded), [&](auto i) {
return inputs[i].strides()[axis] == 0;
});
std::sort(preloaded.begin(), preloaded.end(), by(std::less<>{}, [&](auto i) { std::sort(preloaded.begin(), preloaded.end(), by(std::less<>{}, [&](auto i) {
return inputs[i].bytes(); return inputs[i].bytes();
})); }));
......
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