refactor(aten::squeeze|aten::unsqueeze): lint squeeze and unsqueeze

Signed-off-by: Naren Dasan <[email protected]>
Signed-off-by: Naren Dasan <[email protected]>

Author: narendasan

core/conversion/converters/impl/squeeze.cpp

@@ -22,7 +22,7 @@ auto squeeze_registrations TRTORCH_UNUSED = RegisterNodeConversionPatterns().pat
 
        auto selfDim = util::toVec(self->getDimensions());
        if (dim < 0) {
-           dim = selfDim.size() + dim;
+         dim = selfDim.size() + dim;
        }
 
        auto shuffle_layer = ctx->net->addShuffle(*self);

core/conversion/converters/impl/unsqueeze.cpp

@@ -22,7 +22,7 @@ auto unsqueeze_registrations TRTORCH_UNUSED = RegisterNodeConversionPatterns().p
 
        auto selfDim = util::toVec(self->getDimensions());
        if (dim < 0) {
-           dim = selfDim.size() + dim;
+         dim = selfDim.size() + dim;
        }
 
        auto shuffle_layer = ctx->net->addShuffle(*self);

core/util/trt_util.cpp

@@ -204,11 +204,9 @@ nvinfer1::Dims squeezeDims(const nvinfer1::Dims& d, int pos) {
       if (i <= d.nbDims) {
         dims.d[j] = d.d[i];
       }
-      
     }
     i++;
     j++;
-    
   }
 
   dims.nbDims = d.nbDims - 1;

tests/core/conversion/converters/test_squeeze.cpp

@@ -4,8 +4,6 @@
 #include "tests/util/util.h"
 #include "torch/csrc/jit/ir/irparser.h"
 
-
-
 TEST(Converters, ATenSqueezeConvertsCorrectly) {
   const auto graph = R"IR(
       graph(%0 : Tensor):

tests/core/conversion/converters/test_unsqueeze.cpp

@@ -4,8 +4,6 @@
 #include "tests/util/util.h"
 #include "torch/csrc/jit/ir/irparser.h"
 
-
-
 TEST(Converters, ATenUnsqueezeConvertsCorrectly) {
   const auto graph = R"IR(
       graph(%0 : Tensor):