Skip to content

Commit 7d6ee4e

Browse files
committed
support dropout-10
1 parent 2a6f18c commit 7d6ee4e

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

tf2onnx/onnx_opset/tensor.py

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,13 +59,24 @@ def _wrap_concat_with_cast(ctx, node):
5959
ctx.copy_shape(output_name, output_cast.output[0])
6060

6161

62-
@tf_op(["Size", "Flatten", "Dropout"])
62+
@tf_op(["Size", "Flatten"])
6363
class DirectOp:
6464
@classmethod
6565
def version_4(cls, ctx, node, **kwargs):
6666
pass
6767

6868

69+
@tf_op("Dropout")
70+
class Dropout:
71+
@classmethod
72+
def version_4(cls, ctx, node, **kwargs):
73+
pass
74+
75+
@classmethod
76+
def version_10(cls, ctx, node, **kwargs):
77+
pass
78+
79+
6980
@tf_op("Identity")
7081
class Identity:
7182
@classmethod

0 commit comments

Comments
 (0)