Commit 3c45f2ed authored by Paul's avatar Paul
Browse files

Formatting

parent 9b8d62d1
...@@ -22,18 +22,18 @@ bool is_reshaper(instruction_ref ins) ...@@ -22,18 +22,18 @@ bool is_reshaper(instruction_ref ins)
bool is_transpose_output(instruction_ref ins) bool is_transpose_output(instruction_ref ins)
{ {
if (ins->outputs().size() != 1) if(ins->outputs().size() != 1)
return false; return false;
if (ins->outputs().front()->name() == "contiguous") if(ins->outputs().front()->name() == "contiguous")
return is_transpose_output(ins->outputs().front()); return is_transpose_output(ins->outputs().front());
return ins->outputs().front()->name() == "transpose"; return ins->outputs().front()->name() == "transpose";
} }
instruction_ref find_transpose_input(instruction_ref ins) instruction_ref find_transpose_input(instruction_ref ins)
{ {
if (ins->inputs().size() != 1) if(ins->inputs().size() != 1)
return ins; return ins;
if (ins->inputs().front()->name() == "contiguous") if(ins->inputs().front()->name() == "contiguous")
return find_transpose_input(ins->inputs().front()); return find_transpose_input(ins->inputs().front());
if(ins->inputs().front()->name() == "transpose") if(ins->inputs().front()->name() == "transpose")
return ins->inputs().front(); return ins->inputs().front();
...@@ -78,9 +78,9 @@ void simplify_reshapes::apply(program& p) const ...@@ -78,9 +78,9 @@ void simplify_reshapes::apply(program& p) const
p.replace_instruction(r.first, r.second); p.replace_instruction(r.first, r.second);
} }
} }
else if (ins->name() == "transpose") else if(ins->name() == "transpose")
{ {
if (is_transpose_output(ins)) if(is_transpose_output(ins))
continue; continue;
auto x = ins; auto x = ins;
auto t = ins; auto t = ins;
...@@ -89,7 +89,7 @@ void simplify_reshapes::apply(program& p) const ...@@ -89,7 +89,7 @@ void simplify_reshapes::apply(program& p) const
x = t; x = t;
t = find_transpose_input(x); t = find_transpose_input(x);
} while(x != t and t->name() == "transpose"); } while(x != t and t->name() == "transpose");
if (t == ins or t->name() != "transpose") if(t == ins or t->name() != "transpose")
continue; continue;
p.replace_instruction(ins, t->inputs().front()); p.replace_instruction(ins, t->inputs().front());
} }
......
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