From a3188586af410859501284420e9dfd83d1f73525 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yaman.umuroglu@amd.com>
Date: Fri, 5 Apr 2024 13:29:36 +0200
Subject: [PATCH] Add AveragePoolNHWC to registry

---
 src/qonnx/custom_op/general/__init__.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/qonnx/custom_op/general/__init__.py b/src/qonnx/custom_op/general/__init__.py
index a656d4a5..8aa98936 100644
--- a/src/qonnx/custom_op/general/__init__.py
+++ b/src/qonnx/custom_op/general/__init__.py
@@ -30,7 +30,7 @@
 from qonnx.custom_op.general.debugmarker import DebugMarker
 from qonnx.custom_op.general.genericpartition import GenericPartition
 from qonnx.custom_op.general.im2col import Im2Col
-from qonnx.custom_op.general.maxpoolnhwc import MaxPoolNHWC
+from qonnx.custom_op.general.maxpoolnhwc import MaxPoolNHWC, AveragePoolNHWC
 from qonnx.custom_op.general.multithreshold import MultiThreshold
 from qonnx.custom_op.general.quant import Quant
 from qonnx.custom_op.general.quantavgpool2d import QuantAvgPool2d
@@ -42,6 +42,7 @@
 custom_op["DebugMarker"] = DebugMarker
 custom_op["QuantAvgPool2d"] = QuantAvgPool2d
 custom_op["MaxPoolNHWC"] = MaxPoolNHWC
+custom_op["AveragePoolNHWC"] = AveragePoolNHWC
 custom_op["GenericPartition"] = GenericPartition
 custom_op["MultiThreshold"] = MultiThreshold
 custom_op["XnorPopcountMatMul"] = XnorPopcountMatMul