Skip to content

Commit 143fc3b

Browse files
committed
chore: update annotation in line 19 of pooling.cpp
Signed-off-by: Ruoqian Guo <[email protected]>
1 parent 15aa098 commit 143fc3b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

core/conversion/converters/impl/pooling.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ bool GlobalPoolingConverter(
1616
nvinfer1::PoolingType pool_type) {
1717
auto in = args[0].ITensorOrFreeze(ctx);
1818
nvinfer1::Dims dims = in->getDimensions();
19-
// Generate a bitmask of all 1s except the last 2 bits (N and C axes) when dims.nbDims >= 2
19+
// Generate a bitmask of all 1s except the last 2 bits (N and C axes) when dims.nbDims > 2
2020
uint32_t reduceAxes = ((1 << dims.nbDims) - 1) & ~0b11;
2121
// Generate a bitmask of all 1s except the last 1 bits (N axes) when dims.nbDims == 2. `aten::adaptive_avg_pool1d`'s
2222
// input can be (N, C, L) or (C, L).

0 commit comments

Comments
 (0)