message: [TF:XLA] Add additional test case for tf.gather.
diff: @@ -122,6 +122,20 @@ class GatherTest(xla_test.XLATestCase):
  gather_np = np.take(params, indices, axis=axis)
  self.assertAllEqual(gather_np, gather_value)
  + def testIndicesWithDifferentDimensions(self):
  + with self.test_session():
  + for dtype in self.numeric_tf_types:
  + params = array_ops.placeholder(dtype=dtype)
  + indices = array_ops.placeholder(dtype=np.int32)
  + with self.test_scope():
  + gather = array_ops.gather(params, indices)
  + self.assertAllEqual(
  + 7, gather.eval(feed_dict={params: [4, 7, 2], indices: 1}))
  + self.assertAllEqual(
  + [7], gather.eval(feed_dict={params: [4, 7, 2], indices: [1]}))
  + self.assertAllEqual(
  + [[7]], gather.eval(feed_dict={params: [4, 7, 2], indices: [[1]]}))
  +
  class GatherBenchmark(test.Benchmark):
  """Microbenchmarks for the gather op."""
  