Commit 26df5406 authored by Khalique's avatar Khalique
Browse files

formatting

parent 3acbd087
...@@ -41,7 +41,7 @@ struct tf_parser ...@@ -41,7 +41,7 @@ struct tf_parser
auto attrs = attributes.at(s).list().i(); auto attrs = attributes.at(s).list().i();
std::vector<size_t> axes; std::vector<size_t> axes;
copy(attrs.begin(), attrs.end(), std::back_inserter(axes)); copy(attrs.begin(), attrs.end(), std::back_inserter(axes));
if (is_nhwc) if(is_nhwc)
{ {
for(size_t i = 0; i < axes.size(); ++i) for(size_t i = 0; i < axes.size(); ++i)
{ {
...@@ -78,7 +78,6 @@ struct tf_parser ...@@ -78,7 +78,6 @@ struct tf_parser
default: break; default: break;
} }
} }
} }
tf_parser() tf_parser()
...@@ -286,7 +285,7 @@ struct tf_parser ...@@ -286,7 +285,7 @@ struct tf_parser
op.dilation[1] = dilation[3]; op.dilation[1] = dilation[3];
} }
auto l0 = args[1]; auto l0 = args[1];
if (l0->get_operator().name() == "transpose" and is_nhwc) if(l0->get_operator().name() == "transpose" and is_nhwc)
{ {
l0 = prog.add_instruction(op::transpose{{1, 3, 0, 2}}, args[1]); l0 = prog.add_instruction(op::transpose{{1, 3, 0, 2}}, args[1]);
} }
......
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