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

Raise exception for mismatched argument type

Browse Source 
PiperOrigin-RevId: 251865629
This commit is contained in:
Gaurav Jain 2019-06-06 09:24:50 -07:00 committed by TensorFlower Gardener
parent 0b82e86457
commit 603018d47e
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/python/framework/op_def_library.py
View File

@ -618,7 +618,12 @@ class OpDefLibrary(object):
attr_value = base_types[0] attr_value = base_types[0]
if input_arg.type_attr in attrs: if input_arg.type_attr in attrs:
if attrs[input_arg.type_attr] != attr_value: if attrs[input_arg.type_attr] != attr_value:
assert False, "Unreachable" raise TypeError(
"Input '%s' of '%s' Op has type %s that does not "
"match type %s of argument '%s'." %
(input_name, op_type_name, dtypes.as_dtype(attr_value).name,
dtypes.as_dtype(attrs[input_arg.type_attr]).name,
inferred_from[input_arg.type_attr]))
else: else:
for base_type in base_types: for base_type in base_types:
_SatisfiesTypeConstraint(base_type, _SatisfiesTypeConstraint(base_type,