Commit ee55f2c8 authored by Khalique's avatar Khalique
Browse files

formatting

parent 8bf2940a
......@@ -159,7 +159,8 @@ struct tf_parser
std::size_t axis = args[axis_idx]->eval().at<int64_t>();
op::concat op{axis};
// 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&,
......@@ -236,7 +237,7 @@ struct tf_parser
}
}
auto l0 = args[0];
if (l0->name() == "@param")
if(l0->name() == "@param")
{
if(is_nhwc)
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