Method test_he_normal_gpu

test/python/test_initializer.py:115–116

(self)

    @unittest.skipIf(not singa_wrap.USE_CUDA, 'CUDA is not enabled')
    def test_he_normal_gpu(self):
        self.he_uniform(gpu_dev)

    def test_he_normal_cpu(self):
        self.he_uniform(cpu_dev)