Commit ee55f2c8 authored by Khalique's avatar Khalique
Browse files

formatting

parent 8bf2940a
...@@ -159,7 +159,8 @@ struct tf_parser ...@@ -159,7 +159,8 @@ struct tf_parser
std::size_t axis = args[axis_idx]->eval().at<int64_t>(); std::size_t axis = args[axis_idx]->eval().at<int64_t>();
op::concat op{axis}; op::concat op{axis};
// return only first N arguments (assuming last index is the axis value) // return only first N arguments (assuming last index is the axis value)
return prog.add_instruction(op, std::vector<instruction_ref>(args.begin(), args.begin() + axis)); return prog.add_instruction(
op, std::vector<instruction_ref>(args.begin(), args.begin() + axis));
} }
instruction_ref parse_constant(const std::string&, instruction_ref parse_constant(const std::string&,
...@@ -236,7 +237,7 @@ struct tf_parser ...@@ -236,7 +237,7 @@ struct tf_parser
} }
} }
auto l0 = args[0]; auto l0 = args[0];
if (l0->name() == "@param") if(l0->name() == "@param")
{ {
if(is_nhwc) if(is_nhwc)
l0 = prog.add_instruction(op::transpose{{0, 3, 1, 2}}, l0); l0 = prog.add_instruction(op::transpose{{0, 3, 1, 2}}, l0);
......
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