Commit b2f08d98 authored by Paul's avatar Paul
Browse files

Format

parent 26277ec5
...@@ -65,10 +65,12 @@ vectorize vectorize::elements(std::size_t axis, const std::vector<shape>& inputs ...@@ -65,10 +65,12 @@ vectorize vectorize::elements(std::size_t axis, const std::vector<shape>& inputs
return 1; return 1;
if(len == 1 and input.elements() > sizes.front()) if(len == 1 and input.elements() > sizes.front())
return sizes.front(); return sizes.front();
auto it = std::find_if( auto it = std::find_if(sizes.begin(), sizes.end(), [&](auto vsize) {
sizes.begin(), sizes.end(), [&](auto vsize) { // The len is divisible by the size and all the strides are divisible by
// The len is divisible by the size and all the strides are divisible by the size // the size
return (len % vsize) == 0 and std::all_of(input.strides().begin(), input.strides().end(), [&](auto i) { return (len % vsize) == 0 and
std::all_of(
input.strides().begin(), input.strides().end(), [&](auto i) {
return contains({0, 1}, i) or i % vsize == 0; return contains({0, 1}, i) or i % vsize == 0;
}); });
}); });
......
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