Skip to content

GitLab

Unverified Commit f2f085bf authored by Francisco Massa's avatar Francisco Massa Committed by GitHub
Browse files

Fix C++ lint (#2041)

parent 7499851f
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
torchvision/csrc/cpu/DeformConv_cpu.cpp
View file @ f2f085bf
...@@ -392,8 +392,8 @@ at::Tensor DeformConv2d_forward_cpu( ...@@ -392,8 +392,8 @@ at::Tensor DeformConv2d_forward_cpu(
.addmm_(weight[g].flatten(1), columns[g]) .addmm_(weight[g].flatten(1), columns[g])
.view_as(out_buf[b][g]); .view_as(out_buf[b][g]);
} }
columns = columns.view( columns =
{columns.size(0) * columns.size(1), columns.size(2)}); columns.view({columns.size(0) * columns.size(1), columns.size(2)});
} }
out_buf = out_buf.view({batch_sz / n_parallel_imgs, out_buf = out_buf.view({batch_sz / n_parallel_imgs,
......
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