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:
parent
8b935680a5
commit
7cbf6afd91
@ -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()
|
||||
|
||||
|
@ -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:
|
||||
|
Loading…
x
Reference in New Issue
Block a user