diff --git a/paddle/fluid/inference/tensorrt/convert/shuffle_channel_op.cc b/paddle/fluid/inference/tensorrt/convert/shuffle_channel_op.cc
index 0f891e0f9f4e6731199e4a6884ec74a1265b3fef..471f39597477575c69dfa72bfc2159b3e6520723 100644
--- a/paddle/fluid/inference/tensorrt/convert/shuffle_channel_op.cc
+++ b/paddle/fluid/inference/tensorrt/convert/shuffle_channel_op.cc
@@ -29,7 +29,12 @@ class ShuffleChannelOpConverter : public OpConverter {
     // Declare inputs
     auto* input = engine_->GetITensor(op_desc.Input("X")[0]);
     auto input_dims = input->getDimensions();
-    PADDLE_ENFORCE(input_dims.nbDims == 3);
+    PADDLE_ENFORCE_EQ(
+        input_dims.nbDims, 3,
+        platform::errors::InvalidArgument("ShuffleChannel TRT op converter "
+                                          "input dims is invalid. The input "
+                                          "dims size should be 3, but got %d.",
+                                          input_dims.nbDims));
     int c = input_dims.d[0];
     int h = input_dims.d[1];
     int w = input_dims.d[2];