aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/framework
diff options
context:
space:
mode:
authorGravatar Skye Wanderman-Milne <skyewm@google.com>2018-03-09 14:00:23 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-03-09 14:05:00 -0800
commit381f000714b3ecd4be79f057c7a230d5152fa3dd (patch)
tree0e81a8299f4ee558897d3a51c956824dfb59b149 /tensorflow/contrib/framework
parenta60c7785325fce1d6c0c388c7ab348ac228a8032 (diff)
Add smart_case that calls smart_cond.
PiperOrigin-RevId: 188534066
Diffstat (limited to 'tensorflow/contrib/framework')
-rw-r--r--tensorflow/contrib/framework/__init__.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/framework/__init__.py b/tensorflow/contrib/framework/__init__.py
index 21f9651318..3398b3fd1c 100644
--- a/tensorflow/contrib/framework/__init__.py
+++ b/tensorflow/contrib/framework/__init__.py
@@ -90,6 +90,7 @@ See the @{$python/contrib.framework} guide.
@@smart_cond
@@smart_constant_value
+@@smart_case
@@CriticalSection
@@ -108,6 +109,7 @@ from tensorflow.contrib.framework.python.ops import *
from tensorflow.python.framework.ops import prepend_name_scope
from tensorflow.python.framework.ops import strip_name_scope
+from tensorflow.python.framework.smart_cond import smart_case
from tensorflow.python.framework.smart_cond import smart_cond
from tensorflow.python.framework.smart_cond import smart_constant_value
from tensorflow.python.framework.tensor_spec import BoundedTensorSpec