Skip to content

Commit fee7ba3

Browse files
committed
Modifying style things
Signed-off-by: Ruoqian Guo <[email protected]>
1 parent 327191c commit fee7ba3

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

core/conversion/evaluators/aten.cpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -182,14 +182,16 @@ auto aten_registrations TRTORCH_UNUSED =
182182
auto tensor = tensor_var.ITensor();
183183
auto dims = util::toVec(tensor->getDimensions());
184184
auto nbDims = tensor->getDimensions().nbDims;
185-
if (dim < 0)
185+
if (dim < 0) {
186186
dim += nbDims;
187+
}
187188
return dims[dim];
188189
} else {
189190
auto tensor = tensor_var.unwrapToTensor();
190191
auto nbDims = tensor.sizes().size();
191-
if (dim < 0)
192+
if (dim < 0) {
192193
dim += nbDims;
194+
}
193195
return tensor.sizes()[dim];
194196
}
195197
}

0 commit comments

Comments
 (0)