tf.disable_eager_execution is now tf.compat.v1.disable_eager_execution

tf.reset_default_graph is now tf.compat.v1.reset_default_graph

PiperOrigin-RevId: 275543239
Change-Id: Ia4ff22131889ae49d51b85fce2654ae146a89444
This commit is contained in:
Brian Atkinson 2019-10-18 14:09:00 -07:00 committed by TensorFlower Gardener
parent 8b935680a5
commit 7cbf6afd91
2 changed files with 2 additions and 2 deletions

View File

@ -101,7 +101,7 @@ bin_path = None
def main(unused_args):
# Eager execution is enabled by default in TF 2.0, but generated example
# tests are still using non-eager features (e.g. `tf.placeholder`).
tf.disable_eager_execution()
tf.compat.v1.disable_eager_execution()
options = generate_examples_lib.Options()

View File

@ -406,7 +406,7 @@ def make_zip_of_tests(options,
# Build graph
report["tf_log"] = ""
report["toco_log"] = ""
tf.reset_default_graph()
tf.compat.v1.reset_default_graph()
with tf.device("/cpu:0"):
try: