Skip to content

GitLab

Commit 7ccdb25e authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent 5a9bb616
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
src/targets/cpu/lowering.cpp
View file @ 7ccdb25e
...@@ -654,9 +654,9 @@ struct cpu_argmax ...@@ -654,9 +654,9 @@ struct cpu_argmax
argument compute(context&, const shape& output_shape, std::vector<argument> args) const argument compute(context&, const shape& output_shape, std::vector<argument> args) const
{ {
argument result{output_shape}; argument result{output_shape};
auto batch_lens = args.front().get_shape().lens(); auto batch_lens = args.front().get_shape().lens();
std::size_t batch_item_num = batch_lens[op.axis]; std::size_t batch_item_num = batch_lens[op.axis];
batch_lens[op.axis] = 1; batch_lens[op.axis] = 1;
shape batch_shape{shape::int32_type, batch_lens}; shape batch_shape{shape::int32_type, batch_lens};
result.visit([&](auto output) { result.visit([&](auto output) {
...@@ -700,9 +700,9 @@ struct cpu_argmin ...@@ -700,9 +700,9 @@ struct cpu_argmin
argument compute(context&, const shape& output_shape, std::vector<argument> args) const argument compute(context&, const shape& output_shape, std::vector<argument> args) const
{ {
argument result{output_shape}; argument result{output_shape};
auto batch_lens = args.front().get_shape().lens(); auto batch_lens = args.front().get_shape().lens();
std::size_t batch_item_num = batch_lens[op.axis]; std::size_t batch_item_num = batch_lens[op.axis];
batch_lens[op.axis] = 1; batch_lens[op.axis] = 1;
shape batch_shape{shape::int32_type, batch_lens}; shape batch_shape{shape::int32_type, batch_lens};
result.visit([&](auto output) { result.visit([&](auto output) {
......
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