diff --git a/onnx2trt_utils.cpp b/onnx2trt_utils.cpp index b407000..10149de 100644 --- a/onnx2trt_utils.cpp +++ b/onnx2trt_utils.cpp @@ -1509,6 +1509,8 @@ NodeImportResult poolingHelper(IImporterContext* ctx, ::ONNX_NAMESPACE::NodeProt } nvinfer1::IPoolingLayer* poolingLayer = ctx->network()->addPoolingNd(*tensorPtr, type, kernel_size); + ASSERT(poolingLayer && "Failed to add pooling layer.", ErrorCode::kUNSUPPORTED_NODE); + poolingLayer->setStrideNd(strides); // This member is ignored in maxpooling poolingLayer->setAverageCountExcludesPadding(exclude_padding);