"git@developer.sourcefind.cn:gaoqiong/migraphx.git" did not exist on "75d5c66003a2248b08ddb3a1b10c97e045267a08"
Commit b02a8685 authored by Khalique's avatar Khalique
Browse files

formatting

parent f958d258
...@@ -43,7 +43,9 @@ struct tf_parser ...@@ -43,7 +43,9 @@ struct tf_parser
copy(attrs.begin(), attrs.end(), std::back_inserter(axes)); copy(attrs.begin(), attrs.end(), std::back_inserter(axes));
if(is_nhwc) if(is_nhwc)
{ {
std::transform(axes.begin(), axes.end(), axes.begin(), [&](size_t axis) { return parse_axis(axis); }); std::transform(axes.begin(), axes.end(), axes.begin(), [&](size_t axis) {
return parse_axis(axis);
});
} }
return axes; return axes;
} }
...@@ -392,8 +394,9 @@ struct tf_parser ...@@ -392,8 +394,9 @@ struct tf_parser
return prog.add_instruction(op::reshape{{long(dims[0]), long(dims[1])}}, s); return prog.add_instruction(op::reshape{{long(dims[0]), long(dims[1])}}, s);
} }
instruction_ref instruction_ref parse_squeeze(const std::string&,
parse_squeeze(const std::string&, const attribute_map& attributes, std::vector<instruction_ref> args) const attribute_map& attributes,
std::vector<instruction_ref> args)
{ {
op::squeeze op; op::squeeze op;
auto axes = parse_axes(attributes, "squeeze_dims"); auto axes = parse_axes(attributes, "squeeze_dims");
......
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