Skip to content

Commit 71e34df

Browse files
committed
fix Noop unit test
1 parent c7d3640 commit 71e34df

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

tf2onnx/onnx_opset/misc.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,8 @@
1818

1919
# pylint: disable=unused-argument,missing-docstring
2020

21-
@tf_op(["CheckNumerics", "StopGradient", "NoOp"])
22-
class NukeNode:
21+
@tf_op(["CheckNumerics", "StopGradient"])
22+
class MoveToIdent:
2323
@classmethod
2424
def version_4(cls, ctx, node, **kwargs):
2525
node.type = "Identity"
@@ -39,3 +39,10 @@ class DirectOp:
3939
@classmethod
4040
def version_4(cls, ctx, node, **kwargs):
4141
pass
42+
43+
44+
@tf_op("NoOp")
45+
class NukeNode:
46+
@classmethod
47+
def version_4(cls, ctx, node, **kwargs):
48+
ctx.remove_node(node.name)

0 commit comments

Comments
 (0)