(self)
| 40 | self.assertAllEqual([[4.]], y.numpy()) |
| 41 | |
| 42 | def testInstantError(self): |
| 43 | if test_util.is_gpu_available(): |
| 44 | # TODO(nareshmodi): make this test better |
| 45 | self.skipTest("Gather doesn't do index checking on GPUs") |
| 46 | |
| 47 | with self.assertRaisesRegexp(errors.InvalidArgumentError, |
| 48 | r'indices = 7 is not in \[0, 3\)'): |
| 49 | array_ops.gather([0, 1, 2], 7) |
| 50 | |
| 51 | def testGradients(self): |
| 52 |