diff --git a/tensorflow/lite/python/op_hint.py b/tensorflow/lite/python/op_hint.py index c8a031ca17a..bd62ab7250d 100644 --- a/tensorflow/lite/python/op_hint.py +++ b/tensorflow/lite/python/op_hint.py @@ -728,10 +728,10 @@ def _find_all_hints_in_nodes(nodes): for node in nodes: attr = node.attr # This is an op hint if it has a FUNCTION_UUID_ATTR, otherwise skip - uuid = attr[OpHint.FUNCTION_UUID_ATTR].s if (OpHint.FUNCTION_UUID_ATTR not in attr or not attr[OpHint.FUNCTION_UUID_ATTR].s): continue + uuid = attr[OpHint.FUNCTION_UUID_ATTR].s # Start building function call_def = func_calls[uuid]