Tied a few loose ends:

benoitsteiner · tensorflower-gardener · commit fe70d8fff854 · 2016-06-16T11:33:42.000-07:00
* Added a test to ensure that we can enqueue / dequeue tensors of half floats
 * Fixed the description of add_check_numerics_ops
Change: 125082343
diff --git a/tensorflow/python/kernel_tests/fifo_queue_test.py b/tensorflow/python/kernel_tests/fifo_queue_test.py
@@ -90,6 +90,12 @@ def testEnqueue(self):
       enqueue_op = q.enqueue((10.0,))
       enqueue_op.run()
 
+  def testEnqueueHalf(self):
+    with self.test_session():
+      q = tf.FIFOQueue(10, tf.float16)
+      enqueue_op = q.enqueue((10.0,))
+      enqueue_op.run()
+
   def testEnqueueWithShape(self):
     with self.test_session():
       q = tf.FIFOQueue(10, tf.float32, shapes=(3, 2))
@@ -174,6 +180,20 @@ def testDequeue(self):
         vals = dequeued_t.eval()
         self.assertEqual([elems[i]], vals)
 
+  def testDequeueHalf(self):
+    with self.test_session():
+      q = tf.FIFOQueue(10, tf.float16)
+      elems = [10.0, 20.0, 30.0]
+      enqueue_ops = [q.enqueue((x,)) for x in elems]
+      dequeued_t = q.dequeue()
+
+      for enqueue_op in enqueue_ops:
+        enqueue_op.run()
+
+      for i in xrange(len(elems)):
+        vals = dequeued_t.eval()
+        self.assertEqual([elems[i]], vals)
+
   def testEnqueueAndBlockingDequeue(self):
     with self.test_session() as sess:
       q = tf.FIFOQueue(3, tf.float32)
diff --git a/tensorflow/python/ops/numerics.py b/tensorflow/python/ops/numerics.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 # ==============================================================================
 
-"""Connects all float and double tensors to CheckNumericsOp."""
+"""Connects all half, float and double tensors to CheckNumericsOp."""
 
 from __future__ import absolute_import
 from __future__ import division
@@ -47,10 +47,10 @@ def verify_tensor_all_finite(t, msg, name=None):
 def add_check_numerics_ops():
   """Connect a `check_numerics` to every floating point tensor.
 
-  `check_numerics` operations themselves are added for each `float` or `double`
-  tensor in the graph. For all ops in the graph, the `check_numerics` op for
-  all of its (`float` or `double`) inputs is guaranteed to run before the
-  `check_numerics` op on any of its outputs.
+  `check_numerics` operations themselves are added for each `half`, `float`,
+  or `double` tensor in the graph. For all ops in the graph, the
+  `check_numerics` op for all of its (`half`, `float`, or `double`) inputs
+  is guaranteed to run before the `check_numerics` op on any of its outputs.
 
   Returns:
     A `group` op depending on all `check_numerics` ops added.
