Skip to content

Commit 71e7fa3

Browse files
author
Trevor Morris
committed
Fix small bug preventing TRT runtime compilation for versions < 6
1 parent 3734d5f commit 71e7fa3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/runtime/contrib/tensorrt/tensorrt_ops.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -447,7 +447,7 @@ class BatchNormOpConverter : public TensorRTOpConverter {
447447
nvinfer1::IScaleLayer* scale_layer = params->network->addScaleNd(
448448
*input, nvinfer1::ScaleMode::kCHANNEL, weight_shift, weight_scale, power, channel_dim);
449449
#else
450-
ICHECK_EQ(input->getDimensions().nbDims(), 3);
450+
ICHECK_EQ(input->getDimensions().nbDims, 3);
451451
nvinfer1::IScaleLayer* scale_layer = params->network->addScale(
452452
*input, nvinfer1::ScaleMode::kCHANNEL, weight_shift, weight_scale, power);
453453
#endif

0 commit comments

Comments
 (0)