diff --git a/python/oneflow/nn/functional/__init__.py b/python/oneflow/nn/functional/__init__.py
index 998ba700629..4bdacc41992 100644
--- a/python/oneflow/nn/functional/__init__.py
+++ b/python/oneflow/nn/functional/__init__.py
@@ -18,6 +18,12 @@
 from oneflow.F import conv1d
 from oneflow.F import conv2d
 from oneflow.F import conv3d
+from oneflow.F import avgpool_1d
+from oneflow.F import avgpool_2d
+from oneflow.F import avgpool_3d
+from oneflow.F import maxpool_1d
+from oneflow.F import maxpool_2d
+from oneflow.F import maxpool_3d
 from oneflow.F import adaptive_avg_pool1d
 from oneflow.F import adaptive_avg_pool2d
 from oneflow.F import adaptive_avg_pool3d