Add AveragePoolNHWC to registry

fastmachinelearning · Apr 5, 2024 · a318858 · a318858
1 parent a0d4a0a
commit a318858
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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