提交 5eb34a3c authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Remove PoolingMode_t

上级 3b0c4946
...@@ -15,7 +15,6 @@ from aesara.graph.basic import Apply, Constant, Variable ...@@ -15,7 +15,6 @@ from aesara.graph.basic import Apply, Constant, Variable
from aesara.graph.utils import MethodNotDefined from aesara.graph.utils import MethodNotDefined
from aesara.link.c.op import OpenMPOp from aesara.link.c.op import OpenMPOp
from aesara.link.c.params_type import ParamsType from aesara.link.c.params_type import ParamsType
from aesara.link.c.type import EnumList
from aesara.scalar import bool as bool_t from aesara.scalar import bool as bool_t
from aesara.tensor.type import TensorType, int_dtypes from aesara.tensor.type import TensorType, int_dtypes
...@@ -283,18 +282,6 @@ def pool_3d( ...@@ -283,18 +282,6 @@ def pool_3d(
return output return output
# NB: This enum type is currently used in gpuarray/pool.py.
# It may be used later as op param in this current file.
# Enum name and constants names are inspired from cuDNN type `cudnnPoolingMode_t`
# (cf. `aesara/gpuarray/cudnn_defs.py`).
PoolingMode_t = EnumList(
("POOLING_MAX", "max"),
("POOLING_SUM", "sum"),
("POOLING_AVERAGE_COUNT_INCLUDE_PADDING", "average_inc_pad"),
("POOLING_AVERAGE_COUNT_EXCLUDE_PADDING", "average_exc_pad"),
)
class Pool(OpenMPOp): class Pool(OpenMPOp):
""" """
sum or average over different patches. sum or average over different patches.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论