def test_input_uint16(self): self._assert_dtype( np.uint16, tf.uint16, np.matrix([[1, 2], [3, 4]], dtype=np.uint16))