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 7100133 commit b52ee50Copy full SHA for b52ee50
tests/common.py
@@ -326,9 +326,17 @@ 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.
- return True
339
+ # return True
340
341
342
def check_lstm_count(graph, expected_count):
0 commit comments