Skip to content

GitLab

"docs/git@developer.sourcefind.cn:Wenxuan/LightX2V.git" did not exist on "27920492405b23056db3fda4c1dbc0e6d327ac63"
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
argument compute(context&, const shape& output_shape, std::vector<argument> args) const
{
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];
batch_lens[op.axis] = 1;
batch_lens[op.axis] = 1;
shape batch_shape{shape::int32_type, batch_lens};
result.visit([&](auto output) {
......@@ -700,9 +700,9 @@ struct cpu_argmin
argument compute(context&, const shape& output_shape, std::vector<argument> args) const
{
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];
batch_lens[op.axis] = 1;
batch_lens[op.axis] = 1;
shape batch_shape{shape::int32_type, batch_lens};
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