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

Make ConcreteFunction.pretty_printed_signature deterministic.

Browse Source 
PiperOrigin-RevId: 309576809
Change-Id: Iacdba60cd1d0b1d8da887bb41143e89c70bedc43
This commit is contained in:
Edward Loper 2020-05-02 13:50:17 -07:00 committed by TensorFlower Gardener
parent 71e1316f59
commit 14e27e3b9b
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/eager/function.py
View File

@ -22,6 +22,7 @@ from __future__ import print_function
import collections import collections
import functools import functools
import itertools import itertools
import pprint
import threading import threading
import types as types_lib import types as types_lib
import weakref import weakref
@ -2221,7 +2222,8 @@ class ConcreteFunction(object):
pieces = nest.flatten(spec, expand_composites=False) pieces = nest.flatten(spec, expand_composites=False)
markers = [_Marker("<{}>".format(i + 1)) for i in range(len(pieces))] markers = [_Marker("<{}>".format(i + 1)) for i in range(len(pieces))]
structure = nest.pack_sequence_as(spec, markers) structure = nest.pack_sequence_as(spec, markers)
result = "{}".format(structure) # Ensure dictionaries are sorted by key (for determinism)
result = pprint.pformat(structure, width=10000)
for (marker, piece) in zip(markers, pieces): for (marker, piece) in zip(markers, pieces):
result += "\n {}: {}".format(marker, pretty_print_spec(piece)) result += "\n {}: {}".format(marker, pretty_print_spec(piece))
return result return result