Remove v1 decorators from saved_model:saved_model_test.

Browse Source

Saved model tests care about building the right graphs, so they were
forced into graph mode (instead of being ported to tf.function
infrastructure). There are differences in the graphs produced between
v1 and v2, but those can be handled easily - for testing purposes -
by carefully choosing operation names based on the mode.

There were unneeded collection operations in a few tests, which were
removed. Other small changes were made but mostly tests were forced to
run in graph mode.

PiperOrigin-RevId: 324289722
Change-Id: Iad60aac85cc3954755a9c6a35cf5a4ddb42640ed

...

This commit is contained in:

Cesar Crusius 2020-07-31 14:52:21 -07:00 committed by TensorFlower Gardener

parent e3ac0822d8

commit 0bf6bb64ce

1 changed files with 834 additions and 798 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1632

tensorflow/python/saved_model/saved_model_test.py

View File

File diff suppressed because it is too large Load Diff