From 43b572163c3290132efe73d6d1f9c05fa86666f9 Mon Sep 17 00:00:00 2001 From: Francisco Massa Date: Wed, 1 Apr 2020 15:18:02 +0200 Subject: [PATCH] Fix C++ lint --- torchvision/csrc/cpu/DeformConv_cpu.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/torchvision/csrc/cpu/DeformConv_cpu.cpp b/torchvision/csrc/cpu/DeformConv_cpu.cpp index 39273d7ca2c..4b17df5b3de 100644 --- a/torchvision/csrc/cpu/DeformConv_cpu.cpp +++ b/torchvision/csrc/cpu/DeformConv_cpu.cpp @@ -392,8 +392,8 @@ at::Tensor DeformConv2d_forward_cpu( .addmm_(weight[g].flatten(1), columns[g]) .view_as(out_buf[b][g]); } - columns = columns.view( - {columns.size(0) * columns.size(1), columns.size(2)}); + columns = + columns.view({columns.size(0) * columns.size(1), columns.size(2)}); } out_buf = out_buf.view({batch_sz / n_parallel_imgs, @@ -751,7 +751,7 @@ static std::tuple deform_conv2d_backward_input_cpu( weight.size(1), weight.size(2), weight.size(3)}); - + for (int elt = 0; elt < batch_sz / n_parallel_imgs; elt++) { // Separate into weight groups columns = columns.view(