Skip to content

GitLab

"examples/dreambooth/train_dreambooth_inpaint.py" did not exist on "81bdbb5e2a60fdfd4ee3c25ee2619adcd60f96ca"
Commit e42603a5 authored by Paul's avatar Paul
Browse files

Use local shape variable

parent 22c28f75
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
src/propagate_constant.cpp
View file @ e42603a5
...@@ -12,9 +12,10 @@ bool skip_propogate(instruction_ref ins) ...@@ -12,9 +12,10 @@ bool skip_propogate(instruction_ref ins)
{ {
if(ins->name() == "@literal") if(ins->name() == "@literal")
return true; return true;
if(ins->get_shape().broadcasted() and not ins->get_shape().scalar()) auto&& s = ins->get_shape();
if(s.broadcasted() and not s.scalar())
return true; return true;
if(ins->get_shape().scalar() and ins->get_shape().elements() != 1) if(s.scalar() and s.elements() != 1)
return true; return true;
return false; return false;
} }
......
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