From 8d325318210f694b9185a7ce1868a279e67160c3 Mon Sep 17 00:00:00 2001 From: luyang Date: Wed, 25 Aug 2021 17:52:11 +0800 Subject: [PATCH 1/2] export pooling api --- python/oneflow/nn/functional/__init__.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/python/oneflow/nn/functional/__init__.py b/python/oneflow/nn/functional/__init__.py index 998ba700629..37e8b580fc7 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 avg_pool1d +from oneflow.F import avg_pool2d +from oneflow.F import avg_pool3d +from oneflow.F import max_pool1d +from oneflow.F import max_pool2d +from oneflow.F import max_pool3d from oneflow.F import adaptive_avg_pool1d from oneflow.F import adaptive_avg_pool2d from oneflow.F import adaptive_avg_pool3d From 7fa0ef63ccc5efc149ad7b2447560b491f7a04c3 Mon Sep 17 00:00:00 2001 From: luyang Date: Wed, 25 Aug 2021 18:09:12 +0800 Subject: [PATCH 2/2] refine --- python/oneflow/nn/functional/__init__.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/python/oneflow/nn/functional/__init__.py b/python/oneflow/nn/functional/__init__.py index 37e8b580fc7..4bdacc41992 100644 --- a/python/oneflow/nn/functional/__init__.py +++ b/python/oneflow/nn/functional/__init__.py @@ -18,12 +18,12 @@ from oneflow.F import conv1d from oneflow.F import conv2d from oneflow.F import conv3d -from oneflow.F import avg_pool1d -from oneflow.F import avg_pool2d -from oneflow.F import avg_pool3d -from oneflow.F import max_pool1d -from oneflow.F import max_pool2d -from oneflow.F import max_pool3d +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