diff options
| author |  Russell Power <power@google.com> | 2017-11-14 15:57:27 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-14 16:01:28 -0800 |
| commit | 3a63bce95f67854b6745cb43e0e9feb1e93587f1 (patch) | |
| tree | f68c7ba1cc37a6008ad360a197773bb1fd5017d9 | |
| parent | 1bc367859c6dc3c3ab17fad25198f9fb25132e2f (diff) | |
Make the definition of summary operations a warning instead of raising an exception.
PiperOrigin-RevId: 175748972
| -rw-r--r-- | tensorflow/contrib/tpu/python/tpu/tpu.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/contrib/tpu/python/tpu/tpu.py b/tensorflow/contrib/tpu/python/tpu/tpu.py index 9aa5a9c78d..f3ddc09754 100644 --- a/tensorflow/contrib/tpu/python/tpu/tpu.py +++ b/tensorflow/contrib/tpu/python/tpu/tpu.py @@ -29,6 +29,7 @@ from tensorflow.python.framework import ops from tensorflow.python.ops import array_ops from tensorflow.python.ops import control_flow_ops from tensorflow.python.ops import variable_scope +from tensorflow.python.platform import tf_logging as logging _SUMMARY_OPS = ("ScalarSummary",) @@ -111,7 +112,8 @@ class TPUReplicateContext(control_flow_ops.ControlFlowContext): raise ValueError("Placeholder %s is not supported." % op.name) if op.type in _SUMMARY_OPS: - raise ValueError("Summary operations are not currently supported.") + logging.warning( + "Summary operations are not currently supported (%s)" % op.name) if any(x.dtype._is_ref_dtype for x in op.inputs): raise NotImplementedError( |