未验证 提交 7be637a7 编写于 作者: Z zhoutianzi666 提交者: GitHub

[Paddle-TRT] fix_pool (#44139)

* fix_pool

* fix pool_op_plugin
上级 d7be46b3
...@@ -184,11 +184,6 @@ nvinfer1::DimsExprs PoolPluginDynamic::getOutputDimensions( ...@@ -184,11 +184,6 @@ nvinfer1::DimsExprs PoolPluginDynamic::getOutputDimensions(
platform::errors::InvalidArgument( platform::errors::InvalidArgument(
"The Split plugin should be only one input.")); "The Split plugin should be only one input."));
PADDLE_ENFORCE_EQ(
inputs[0].d[1]->isConstant(),
true,
platform::errors::InvalidArgument("The channel dimension should be "
"static, but we found it's dynamic."));
nvinfer1::DimsExprs output(inputs[0]); nvinfer1::DimsExprs output(inputs[0]);
if (is_global_ && !adaptive_) { if (is_global_ && !adaptive_) {
output.d[2] = expr_builder.constant(1); output.d[2] = expr_builder.constant(1);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册