diff options
author | Yifei Feng <yifeif@google.com> | 2017-11-22 00:39:22 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-22 00:42:30 -0800 |
commit | e70c00950d295c519fd9c7f8b12e13a3c5aaf710 (patch) | |
tree | ee1210f8810e0b0fec9346f762e854b371899919 /tensorflow/contrib/lite/testing/generate_examples.py | |
parent | ad7eeec1cc06d7fdba6ee404f03a35fab9cd3e6a (diff) |
Automated g4 rollback of changelist 176615107
PiperOrigin-RevId: 176622438
Diffstat (limited to 'tensorflow/contrib/lite/testing/generate_examples.py')
-rw-r--r-- | tensorflow/contrib/lite/testing/generate_examples.py | 17 |
1 files changed, 6 insertions, 11 deletions
diff --git a/tensorflow/contrib/lite/testing/generate_examples.py b/tensorflow/contrib/lite/testing/generate_examples.py index b122818221..86540d58a6 100644 --- a/tensorflow/contrib/lite/testing/generate_examples.py +++ b/tensorflow/contrib/lite/testing/generate_examples.py @@ -36,10 +36,6 @@ import traceback import zipfile import numpy as np from six import StringIO - -# TODO(aselle): Disable GPU for now -os.environ["CUDA_VISIBLE_DEVICES"] = "-1" - import tensorflow as tf from google.protobuf import text_format # TODO(aselle): switch to TensorFlow's resource_loader @@ -383,13 +379,12 @@ def make_zip_of_tests(zip_path, report["toco_log"] = "" tf.reset_default_graph() - with tf.device('/cpu:0'): - try: - inputs, outputs = make_graph(param_dict_real) - except (tf.errors.UnimplementedError, tf.errors.InvalidArgumentError, - ValueError): - report["tf_log"] += traceback.format_exc() - return None, report + try: + inputs, outputs = make_graph(param_dict_real) + except (tf.errors.UnimplementedError, tf.errors.InvalidArgumentError, + ValueError): + report["tf_log"] += traceback.format_exc() + return None, report sess = tf.Session() try: |