Method test_batchnorm2d_gpu

test/python/test_operation.py:345–346

(self)

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

    def gradients_check(self,
                        func,