update

PaddlePaddle · Apr 23, 2021 · bcf084c · bcf084c · paddle-bot-old · Apr 25, 2021
1 parent 86fd9ab
commit bcf084c
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/paddle/fluid/inference/tensorrt/convert/fc_op.cc b/paddle/fluid/inference/tensorrt/convert/fc_op.cc
@@ -196,11 +196,9 @@ class FcOpConverter : public OpConverter {
       // add shuffle after fc
       nvinfer1::Dims reshape_after_fc_dim;
       reshape_after_fc_dim.nbDims = x_num_col_dims + 1;
-      for (int i = 0; i < x_num_col_dims; i++) {
+      for (int i = 0; i < reshape_after_fc_dim.nbDims; i++) {
         reshape_after_fc_dim.d[i] = 0;
       }
-      auto dim = fc_layer->getOutput(0)->getDimensions();
-      reshape_after_fc_dim.d[x_num_col_dims] = dim.d[x_num_col_dims];
 
       auto* reshape_after_fc_layer =
           TRT_ENGINE_ADD_LAYER(engine_, Shuffle, *fc_layer->getOutput(0));