Commit 20bdf794 authored by Paul's avatar Paul
Browse files

Formatting

parent 22cc51ab
...@@ -114,7 +114,10 @@ inline auto nary(argument result, argument arg1, argument arg2) ...@@ -114,7 +114,10 @@ inline auto nary(argument result, argument arg1, argument arg2)
return [=](auto f) { return [=](auto f) {
// TODO: Check for one broadcast stride // TODO: Check for one broadcast stride
// TODO: Check result and arg1 shape is the same // TODO: Check result and arg1 shape is the same
if(arg1.get_shape().standard() and arg2.get_shape().broadcasted() and std::count_if(arg2.get_shape().strides().begin(), arg2.get_shape().strides().end(), [](auto x) { return x != 0; }) == 1) if(arg1.get_shape().standard() and arg2.get_shape().broadcasted() and
std::count_if(arg2.get_shape().strides().begin(),
arg2.get_shape().strides().end(),
[](auto x) { return x != 0; }) == 1)
{ {
binary_broadcast(result, arg1, arg2)(f); binary_broadcast(result, arg1, arg2)(f);
} }
......
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