We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 15aa098 commit 143fc3bCopy full SHA for 143fc3b
core/conversion/converters/impl/pooling.cpp
@@ -16,7 +16,7 @@ bool GlobalPoolingConverter(
16
nvinfer1::PoolingType pool_type) {
17
auto in = args[0].ITensorOrFreeze(ctx);
18
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
+ // Generate a bitmask of all 1s except the last 2 bits (N and C axes) when dims.nbDims > 2
20
uint32_t reduceAxes = ((1 << dims.nbDims) - 1) & ~0b11;
21
// Generate a bitmask of all 1s except the last 1 bits (N axes) when dims.nbDims == 2. `aten::adaptive_avg_pool1d`'s
22
// input can be (N, C, L) or (C, L).
0 commit comments