diff --git a/core/conversion/converters/impl/plugins/interpolate_plugin.cpp b/core/conversion/converters/impl/plugins/interpolate_plugin.cpp index ae3479f917..2da05a70e5 100755 --- a/core/conversion/converters/impl/plugins/interpolate_plugin.cpp +++ b/core/conversion/converters/impl/plugins/interpolate_plugin.cpp @@ -175,6 +175,8 @@ int InterpolatePlugin::enqueue(const nvinfer1::PluginTensorDesc* inputDesc, cons at::upsample_bilinear2d_out(output, input, {size[0], size[1]}, align_corners); } else if (mode == "trilinear") { at::upsample_trilinear3d_out(output, input, {size[0], size[1], size[2]}, align_corners); + } else if (mode == "adaptive_pool2d") { + at::adaptive_avg_pool2d_out(output, input, {size[0], size[1]}); } cudaEvent_t torch_event;