"docs/source/en/api/pipelines/controlnet_sd3.md" did not exist on "5333f4c0ec1c4a69ad2ada88364c5dd5836ac1b7"
Commit 20bdf794 authored by Paul's avatar Paul
Browse files

Formatting

parent 22cc51ab
...@@ -114,11 +114,14 @@ inline auto nary(argument result, argument arg1, argument arg2) ...@@ -114,11 +114,14 @@ 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);
} }
else else
{ {
nary_impl(result, arg1, arg2)(f); nary_impl(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