Remove reference in docstring to tf.contrib.

PiperOrigin-RevId: 384756222
Change-Id: Ia17e8f2b877e1d055fb04b79d307c060773b0b25
diff --git a/tensorflow/python/ops/control_flow_ops.py b/tensorflow/python/ops/control_flow_ops.py
index 807acf45..e8e73d9 100644
--- a/tensorflow/python/ops/control_flow_ops.py
+++ b/tensorflow/python/ops/control_flow_ops.py
@@ -3389,11 +3389,11 @@
   operation returns the tensors generated by `default`.
 
   `tf.case` supports nested structures as implemented in
-  `tf.contrib.framework.nest`. All of the callables must return the same
-  (possibly nested) value structure of lists, tuples, and/or named tuples.
-  Singleton lists and tuples form the only exceptions to this: when returned by
-  a callable, they are implicitly unpacked to single values. This
-  behavior is disabled by passing `strict=True`.
+  `tf.nest`. All of the callables must return the same (possibly nested) value
+  structure of lists, tuples, and/or named tuples. Singleton lists and tuples
+  form the only exceptions to this: when returned by a callable, they are
+  implicitly unpacked to single values. This behavior is disabled by passing
+  `strict=True`.
 
   @compatibility(v2)
   `pred_fn_pairs` could be a dictionary in v1. However, tf.Tensor and
@@ -3494,11 +3494,11 @@
   operation returns the tensors generated by `default`.
 
   `tf.case` supports nested structures as implemented in
-  `tf.contrib.framework.nest`. All of the callables must return the same
-  (possibly nested) value structure of lists, tuples, and/or named tuples.
-  Singleton lists and tuples form the only exceptions to this: when returned by
-  a callable, they are implicitly unpacked to single values. This
-  behavior is disabled by passing `strict=True`.
+  `tf.nest`. All of the callables must return the same (possibly nested) value
+  structure of lists, tuples, and/or named tuples. Singleton lists and tuples
+  form the only exceptions to this: when returned by a callable, they are
+  implicitly unpacked to single values. This behavior is disabled by passing
+  `strict=True`.
 
   If an unordered dictionary is used for `pred_fn_pairs`, the order of the
   conditional tests is not guaranteed. However, the order is guaranteed to be