Method test_round_gpu

test/python/test_api.py:892–893

(self)

    @unittest.skipIf(not singa_api.USE_CUDA, 'CUDA is not enabled')
    def test_round_gpu(self):
        self._round(gpu_dev)

    def _round(self, dev=gpu_dev):
        x = tensor.Tensor(shape=(3,4,5), device=dev).gaussian(0, 1)