Skip to content

Commit 3169bc1

Browse files
committed
resolve new requested change
1 parent c9d6295 commit 3169bc1

File tree

2 files changed

+2
-6
lines changed

2 files changed

+2
-6
lines changed

tests/common.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@
3434
"group_nodes_by_type",
3535
"test_ms_domain",
3636
"check_node_domain",
37-
"check_thresholded_relu_count"
37+
"check_op_count"
3838
]
3939

4040

@@ -306,10 +306,6 @@ def check_gru_count(graph, expected_count):
306306
return check_op_count(graph, "GRU", expected_count)
307307

308308

309-
def check_thresholded_relu_count(graph, expected_count):
310-
return check_op_count(graph, "ThresholdedRelu", expected_count)
311-
312-
313309
_MAX_MS_OPSET_VERSION = 1
314310

315311

tests/test_backend.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2149,7 +2149,7 @@ def test_thresholded_relu(self):
21492149
x_ = t.call(x)
21502150
_ = tf.identity(x_, name=_TFOUTPUT)
21512151
self._run_test_case([_OUTPUT], {_INPUT: x_val},
2152-
graph_validator=lambda g: check_thresholded_relu_count(g, 1))
2152+
graph_validator=lambda g: check_op_count(g, "ThresholdedRelu", 1))
21532153
tf.reset_default_graph()
21542154

21552155

0 commit comments

Comments
 (0)