From 3a63bce95f67854b6745cb43e0e9feb1e93587f1 Mon Sep 17 00:00:00 2001 From: Russell Power Date: Tue, 14 Nov 2017 15:57:27 -0800 Subject: Make the definition of summary operations a warning instead of raising an exception. PiperOrigin-RevId: 175748972 --- tensorflow/contrib/tpu/python/tpu/tpu.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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( -- cgit v1.2.3