Skip to content

Commit e853156

Browse files
Fixed bug in reading table info from saved models (#1263)
Signed-off-by: Tom Wildenhain <[email protected]>
1 parent 6137189 commit e853156

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

tf2onnx/tf_loader.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -279,13 +279,16 @@ def _get_hash_table_info_from_trackable(trackable, table_names, key_dtypes, valu
279279
removed_resource_to_placeholder, placeholder_to_table_info):
280280
# pylint: disable=protected-access
281281
for r in trackable.__dict__.values():
282-
if isinstance(r, TfRestoredResourceType) and hasattr(r, '_create_resource') and hasattr(r, 'resource_handle'):
282+
if isinstance(r, TfRestoredResourceType) and hasattr(r, '_create_resource'):
283+
try:
284+
table_handle = id(r.resource_handle)
285+
except Exception: # pylint: disable=broad-except
286+
continue
283287
initializer = r._create_resource.concrete_functions[0].function_def
284288
new_names, new_k_dtypes, new_v_dtypes = get_hash_table_info(initializer.node_def)
285289
table_names.extend(new_names)
286290
key_dtypes.extend(new_k_dtypes)
287291
value_dtypes.extend(new_v_dtypes)
288-
table_handle = id(r.resource_handle)
289292
if table_handle in removed_resource_to_placeholder and len(new_names) == 1:
290293
table_info = (new_names[0], new_k_dtypes[0], new_v_dtypes[0])
291294
placeholder_to_table_info[removed_resource_to_placeholder[table_handle]] = table_info

0 commit comments

Comments
 (0)