Commit add40fd1 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent 75c75d77
...@@ -1279,8 +1279,9 @@ struct onnx_parser ...@@ -1279,8 +1279,9 @@ struct onnx_parser
return {hidden_states, last_output, last_cell_output}; return {hidden_states, last_output, last_cell_output};
} }
instruction_ref instruction_ref parse_reduce_sum(const std::string&,
parse_reduce_sum(const std::string&, attribute_map attributes, std::vector<instruction_ref> args) attribute_map attributes,
std::vector<instruction_ref> args)
{ {
std::size_t n_dim = args.front()->get_shape().lens().size(); std::size_t n_dim = args.front()->get_shape().lens().size();
...@@ -1295,12 +1296,12 @@ struct onnx_parser ...@@ -1295,12 +1296,12 @@ struct onnx_parser
} }
int keep_dims = 1; int keep_dims = 1;
if (contains(attributes, "keepdims")) if(contains(attributes, "keepdims"))
{ {
keep_dims = parse_value(attributes.at("keepdims")).at<int>(); keep_dims = parse_value(attributes.at("keepdims")).at<int>();
} }
if (keep_dims == 1) if(keep_dims == 1)
{ {
return prog.add_instruction(op::reduce_sum{axes}, std::move(args)); return prog.add_instruction(op::reduce_sum{axes}, std::move(args));
} }
......
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