refactor(aten::topk): linting topk converter

Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · Feb 2, 2021 · 5bd7528 · 5bd7528
1 parent 22e6a6b
commit 5bd7528
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/core/conversion/converters/impl/topk.cpp b/core/conversion/converters/impl/topk.cpp
@@ -25,11 +25,12 @@ auto topk_registrations TRTORCH_UNUSED = RegisterNodeConversionPatterns().patter
 
        auto selfDim = util::toVec(self->getDimensions());
 
-       //reduceAxes	The reduction dimensions. The bit in position i of bitmask reduceAxes corresponds to explicit dimension i of the result.
-       //E.g., the least significant bit corresponds to the first explicit dimension and the next to least significant bit corresponds to the second explicit dimension.
+       // reduceAxes	The reduction dimensions. The bit in position i of bitmask reduceAxes corresponds to explicit
+       // dimension i of the result. E.g., the least significant bit corresponds to the first explicit dimension and the
+       // next to least significant bit corresponds to the second explicit dimension.
 
        if (dim < 0) {
-           dim = selfDim.size() + dim;
+         dim = selfDim.size() + dim;
        }
 
        uint32_t shiftDim = 1 << dim;