experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Check types of name input parameters

This commit is contained in:
Sergey Mironov 2019-09-03 16:29:08 +03:00 committed by Sergey Mironov
parent a271711455
commit cd8760643c
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/lite/python/util.py
View File

@ -110,6 +110,12 @@ def get_tensors_from_tensor_names(graph, tensor_names):
tensors = [] tensors = []
invalid_tensors = [] invalid_tensors = []
for name in tensor_names: for name in tensor_names:
if not isinstance(name, str):
raise ValueError("Invalid type for a tensor name in the provided graph. "
"Expected type for a tensor name is 'str', instead got "
"type '{}' for tensor name '{}'".format(
type(name), name))
tensor = tensor_name_to_tensor.get(name) tensor = tensor_name_to_tensor.get(name)
if tensor is None: if tensor is None:
invalid_tensors.append(name) invalid_tensors.append(name)