Skip to content

Commit b52ee50

Browse files
committed
reenable lstm count test
1 parent 7100133 commit b52ee50

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

tests/common.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -326,9 +326,17 @@ def group_nodes_by_type(graph):
326326

327327

328328
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
329337
# return len(group_nodes_by_type(graph)[op_type]) == expected_count
330338
# FIXME: after switching to grappler some of the op counts are off. Fix later.
331-
return True
339+
# return True
332340

333341

334342
def check_lstm_count(graph, expected_count):

0 commit comments

Comments
 (0)