Skip to content

Commit 2a6f18c

Browse files
committed
support maxpool-10 and avgpool-10
1 parent a88c048 commit 2a6f18c

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tf2onnx/onnx_opset/nn.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -294,6 +294,14 @@ def version_4(cls, ctx, node, **kwargs):
294294
class PoolOp:
295295
@classmethod
296296
def version_4(cls, ctx, node, **kwargs):
297+
cls._convert(ctx, node, **kwargs)
298+
299+
@classmethod
300+
def version_10(cls, ctx, node, **kwargs):
301+
cls._convert(ctx, node, **kwargs)
302+
303+
@classmethod
304+
def _convert(cls, ctx, node, **kwargs):
297305
# T output = MaxPool(T input, @list(int) ksize, @list(int) strides, @string padding, @string data_format)
298306
# T Y = MaxPool(T X, @AttrType.STRING auto_pad, @AttrType.INTS kernel_shape, @AttrType.INTS pads,
299307
# @AttrType.INTS strides)

0 commit comments

Comments
 (0)