From 52be580585769ee0f11f1b58b9a70e8aee9e033a Mon Sep 17 00:00:00 2001
From: Abhiram Iyer <abhi.iyer.ai@gmail.com>
Date: Thu, 18 Jun 2020 16:29:03 -0700
Subject: [PATCH] feat(): support for adaptive_avg_pool2d plugin Signed-off-by:
 Abhiram Iyer <abhirami@nvidia.com>

Signed-off-by: Abhiram Iyer <abhi.iyer.ai@gmail.com>
---
 core/conversion/converters/impl/plugins/interpolate_plugin.cpp | 2 ++
 1 file changed, 2 insertions(+)

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;