Method test_max_pool_gpu

test/python/test_onnx.py:291–292

(self)

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

    def _batch_norm_helper(self, dev):
        x = np.array([[[[-1, 0, 1]], [[2, 3, 4]]]]).astype(np.float32)