Skip to content

GitLab

"projects/DSVT/dsvt/dsvt.py" did not exist on "456b7403cd9e7572cc47bb8bff6bcc17db882c38"
Commit cffb1b1b authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent 9ea01307
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
src/onnx/onnx.cpp
View file @ cffb1b1b
...@@ -284,7 +284,7 @@ struct onnx_parser ...@@ -284,7 +284,7 @@ struct onnx_parser
keep_dims = parse_value(attributes.at("keepdims")).at<int>(); keep_dims = parse_value(attributes.at("keepdims")).at<int>();
} }
if (keep_dims == 0) if(keep_dims == 0)
{ {
auto ins = prog.add_instruction(op::argmax{axis}, std::move(args)); auto ins = prog.add_instruction(op::argmax{axis}, std::move(args));
return prog.add_instruction(op::squeeze{{static_cast<int64_t>(axis)}}, ins); return prog.add_instruction(op::squeeze{{static_cast<int64_t>(axis)}}, ins);
...@@ -311,7 +311,7 @@ struct onnx_parser ...@@ -311,7 +311,7 @@ struct onnx_parser
keep_dims = parse_value(attributes.at("keepdims")).at<int>(); keep_dims = parse_value(attributes.at("keepdims")).at<int>();
} }
if (keep_dims == 0) if(keep_dims == 0)
{ {
auto ins = prog.add_instruction(op::argmin{axis}, std::move(args)); auto ins = prog.add_instruction(op::argmin{axis}, std::move(args));
return prog.add_instruction(op::squeeze{{static_cast<int64_t>(axis)}}, ins); return prog.add_instruction(op::squeeze{{static_cast<int64_t>(axis)}}, ins);
......
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