提交 bb17cc75 编写于 作者: Z zlsh80826

remove debug message

上级 b62406bc
...@@ -15,8 +15,6 @@ limitations under the License. */ ...@@ -15,8 +15,6 @@ limitations under the License. */
#include "paddle/fluid/inference/tensorrt/convert/op_converter.h" #include "paddle/fluid/inference/tensorrt/convert/op_converter.h"
#include "paddle/fluid/inference/tensorrt/plugin/stack_op_plugin.h" #include "paddle/fluid/inference/tensorrt/plugin/stack_op_plugin.h"
#include <iostream>
namespace paddle { namespace paddle {
namespace inference { namespace inference {
namespace tensorrt { namespace tensorrt {
...@@ -40,10 +38,6 @@ class StackOpConverter : public OpConverter { ...@@ -40,10 +38,6 @@ class StackOpConverter : public OpConverter {
inputs[i] = engine_->GetITensor(input[i]); 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")); int axis = BOOST_GET_CONST(int, op_desc.GetAttr("axis"));
if (axis < 0) { if (axis < 0) {
axis = axis + inputs[0]->getDimensions().nbDims + 1; axis = axis + inputs[0]->getDimensions().nbDims + 1;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册