message: Test more types in ternary_ops_test.
diff: @@ -69,40 +69,41 @@ class TernaryOpsTest(XLATestCase):
  expected=np.array([1, 3, 5], dtype=np.int32))
  def testSelect(self):
  + for dtype in self.numeric_types:
  self._testTernary(
  array_ops.where,
  np.array(0, dtype=np.bool),
  - np.array(2, dtype=np.float32),
  - np.array(7, dtype=np.float32),
  - expected=np.array(7, dtype=np.float32))
  + np.array(2, dtype=dtype),
  + np.array(7, dtype=dtype),
  + expected=np.array(7, dtype=dtype))
  self._testTernary(
  array_ops.where,
  np.array(1, dtype=np.bool),
  - np.array([1, 2, 3, 4], dtype=np.float32),
  - np.array([5, 6, 7, 8], dtype=np.float32),
  - expected=np.array([1, 2, 3, 4], dtype=np.float32))
  + np.array([1, 2, 3, 4], dtype=dtype),
  + np.array([5, 6, 7, 8], dtype=dtype),
  + expected=np.array([1, 2, 3, 4], dtype=dtype))
  self._testTernary(
  array_ops.where,
  np.array(0, dtype=np.bool),
  - np.array([[1, 2], [3, 4], [5, 6]], dtype=np.float32),
  - np.array([[7, 8], [9, 10], [11, 12]], dtype=np.float32),
  - expected=np.array([[7, 8], [9, 10], [11, 12]], dtype=np.float32))
  + np.array([[1, 2], [3, 4], [5, 6]], dtype=dtype),
  + np.array([[7, 8], [9, 10], [11, 12]], dtype=dtype),
  + expected=np.array([[7, 8], [9, 10], [11, 12]], dtype=dtype))
  self._testTernary(
  array_ops.where,
  np.array([0, 1, 1, 0], dtype=np.bool),
  - np.array([1, 2, 3, 4], dtype=np.float32),
  - np.array([5, 6, 7, 8], dtype=np.float32),
  - expected=np.array([5, 2, 3, 8], dtype=np.float32))
  + np.array([1, 2, 3, 4], dtype=dtype),
  + np.array([5, 6, 7, 8], dtype=dtype),
  + expected=np.array([5, 2, 3, 8], dtype=dtype))
  self._testTernary(
  array_ops.where,
  np.array([0, 1, 0], dtype=np.bool),
  - np.array([[1, 2], [3, 4], [5, 6]], dtype=np.float32),
  - np.array([[7, 8], [9, 10], [11, 12]], dtype=np.float32),
  - expected=np.array([[7, 8], [3, 4], [11, 12]], dtype=np.float32))
  + np.array([[1, 2], [3, 4], [5, 6]], dtype=dtype),
  + np.array([[7, 8], [9, 10], [11, 12]], dtype=dtype),
  + expected=np.array([[7, 8], [3, 4], [11, 12]], dtype=dtype))
  def testSlice(self):
  for dtype in self.numeric_types:
  