(self, dev=gpu_dev)
| 520 | |
| 521 | @unittest.skipIf(not singa_api.USE_CUDA, 'CUDA is not enabled') |
| 522 | def test_gaussian_gpu(self, dev=gpu_dev): |
| 523 | x = tensor.Tensor((3, 5, 3, 5), device=dev) |
| 524 | x.gaussian(0, 1) |
| 525 | x = tensor.Tensor((4, 5, 3, 2), device=dev) |
| 526 | x.gaussian(0, 1) |
| 527 | |
| 528 | def _kfloat32_int(self, dev=gpu_dev): |
| 529 | np.random.seed(0) |