diff options
Diffstat (limited to 'tensorflow/python/autograph/pyct/inspect_utils.py')
-rw-r--r-- | tensorflow/python/autograph/pyct/inspect_utils.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/python/autograph/pyct/inspect_utils.py b/tensorflow/python/autograph/pyct/inspect_utils.py index eef74599a7..1416988ea3 100644 --- a/tensorflow/python/autograph/pyct/inspect_utils.py +++ b/tensorflow/python/autograph/pyct/inspect_utils.py @@ -30,10 +30,14 @@ from tensorflow.python.util import tf_inspect def isbuiltin(f): + """Returns True if the argument is a built-in function.""" # Note these return false for isinstance(f, types.BuiltinFunctionType) so we # need to specifically check for them. if f in (range, int, float): return True + if six.PY2: + if f in (xrange,): + return True if isinstance(f, types.BuiltinFunctionType): return True if tf_inspect.isbuiltin(f): |