Skip to content

Commit 965e988

Browse files
committed
fix ut
1 parent dcc6a4f commit 965e988

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_graph.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -249,8 +249,8 @@ def test_relu6(self):
249249
_ = tf.identity(x_, name="output")
250250
g = process_tf_graph(sess.graph, opset=self.config.opset)
251251
self.assertEqual(
252-
'digraph { input1 [op_type=Placeholder shape="[2, 3]"] Relu6 [op_type=Relu] Relu6__2 [op_type=Clip] '
253-
'output [op_type=Identity] input1:0 -> Relu6 Relu6:0 -> Relu6__2 Relu6__2:0 -> output }',
252+
'digraph { input1 [op_type=Placeholder shape="[2, 3]"] Relu6 [op_type=Clip] output [op_type=Identity] '
253+
'input1:0 -> Relu6 Relu6:0 -> output }',
254254
onnx_to_graphviz(g))
255255

256256
def test_conv2d(self):

0 commit comments

Comments
 (0)