diff --git a/paddle/fluid/inference/tensorrt/convert/stack_op.cc b/paddle/fluid/inference/tensorrt/convert/stack_op.cc index c01b87049354af7e06edcbccea4c6c4cc455a6c6..f35024529c61a253f314e5eca985713227d3f343 100644 --- a/paddle/fluid/inference/tensorrt/convert/stack_op.cc +++ b/paddle/fluid/inference/tensorrt/convert/stack_op.cc @@ -15,8 +15,6 @@ limitations under the License. */ #include "paddle/fluid/inference/tensorrt/convert/op_converter.h" #include "paddle/fluid/inference/tensorrt/plugin/stack_op_plugin.h" -#include - namespace paddle { namespace inference { namespace tensorrt { @@ -40,10 +38,6 @@ class StackOpConverter : public OpConverter { inputs[i] = engine_->GetITensor(input[i]); } - auto idim = inputs[0]->getDimensions(); - std::cerr << "Stack input: " << idim.nbDims << " " << idim.d[0] << " " - << idim.d[1] << " " << idim.d[2] << std::endl; - int axis = BOOST_GET_CONST(int, op_desc.GetAttr("axis")); if (axis < 0) { axis = axis + inputs[0]->getDimensions().nbDims + 1;