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(