Skip to content

Commit 5bd7528

Browse files
committed
refactor(aten::topk): linting topk converter
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent 22e6a6b commit 5bd7528

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

core/conversion/converters/impl/topk.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,11 +25,12 @@ auto topk_registrations TRTORCH_UNUSED = RegisterNodeConversionPatterns().patter
2525

2626
auto selfDim = util::toVec(self->getDimensions());
2727

28-
//reduceAxes The reduction dimensions. The bit in position i of bitmask reduceAxes corresponds to explicit dimension i of the result.
29-
//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.
28+
// reduceAxes The reduction dimensions. The bit in position i of bitmask reduceAxes corresponds to explicit
29+
// dimension i of the result. E.g., the least significant bit corresponds to the first explicit dimension and the
30+
// next to least significant bit corresponds to the second explicit dimension.
3031

3132
if (dim < 0) {
32-
dim = selfDim.size() + dim;
33+
dim = selfDim.size() + dim;
3334
}
3435

3536
uint32_t shiftDim = 1 << dim;

0 commit comments

Comments
 (0)