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