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

Fix weakref unwrapping in AutoGraph conversion

Browse Source 
PiperOrigin-RevId: 225850437
This commit is contained in:
A. Unique TensorFlower 2018-12-17 10:37:49 -08:00 committed by TensorFlower Gardener
parent a1c66b9f15
commit d914628367
1 changed files with 5 additions and 0 deletions

5
tensorflow/python/autograph/core/converter.py
View File

@ -218,6 +218,11 @@ class ConversionOptions(object):
def as_qualified_name(o): def as_qualified_name(o):
name = inspect_utils.getqualifiedname(ctx.info.namespace, o, max_depth=1) name = inspect_utils.getqualifiedname(ctx.info.namespace, o, max_depth=1)
if not name: if not name:
if isinstance(o, weakref.ref):
# `o` might already be a weak reference, if this object was
# constructed from code generated by `to_ast` itself.
# If so, unpack it.
o = o()
# TODO(mdan): This needs to account for the symbols defined locally. # TODO(mdan): This needs to account for the symbols defined locally.
name = ctx.namer.new_symbol(o.__name__, ()) name = ctx.namer.new_symbol(o.__name__, ())
ctx.program.add_symbol(name, weakref.ref(o)) ctx.program.add_symbol(name, weakref.ref(o))