We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b52ee50 commit c5872a6Copy full SHA for c5872a6
tests/common.py
@@ -326,17 +326,7 @@ def group_nodes_by_type(graph):
326
327
328
def check_op_count(graph, op_type, expected_count):
329
- actual_count = len(group_nodes_by_type(graph)[op_type])
330
- count_match = actual_count == expected_count
331
- '''
332
- if False == count_match:
333
- print ("LSTM count mismatch:", actual_count, expected_count)
334
- input()
335
336
- return actual_count == expected_count
337
- # return len(group_nodes_by_type(graph)[op_type]) == expected_count
338
- # FIXME: after switching to grappler some of the op counts are off. Fix later.
339
- # return True
+ return len(group_nodes_by_type(graph)[op_type]) == expected_count
340
341
342
def check_lstm_count(graph, expected_count):
0 commit comments