aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/framework/ops.py
diff options
context:
space:
mode:
authorGravatar Sherry Moore <sherrym@google.com>2016-09-13 09:08:10 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-09-13 10:17:12 -0700
commite91ea774cfce323ef5812274f92fa2f13d2f3637 (patch)
tree03aafc080c1ac79e284577cb4a484f096da984c6 /tensorflow/python/framework/ops.py
parentdb37a626fa81fe613aeaa7539e7730b5caaacce6 (diff)
Added to_proto and from_proto for CondContext and WhileContext.
Change: 133013495
Diffstat (limited to 'tensorflow/python/framework/ops.py')
-rw-r--r--tensorflow/python/framework/ops.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/python/framework/ops.py b/tensorflow/python/framework/ops.py
index 8cd5b0fef1..caa9695c8a 100644
--- a/tensorflow/python/framework/ops.py
+++ b/tensorflow/python/framework/ops.py
@@ -3948,6 +3948,10 @@ class GraphKeys(object):
SUMMARY_OP = "summary_op"
GLOBAL_STEP = "global_step"
+ # Key for control flow context.
+ COND_CONTEXT = "cond_context"
+ WHILE_CONTEXT = "while_context"
+
def add_to_collection(name, value):
"""Wrapper for `Graph.add_to_collection()` using the default graph.