def test_pool_py_interface(self): shp = (2, 2, 2, 2) inp = aesara.shared(rand(*shp), "a") inp = aet.as_tensor_variable(inp) with pytest.raises(ValueError): # test when pad >= ws ds_op = GpuPool(ignore_border=True, ndim=2) ds_op(inp, [2, 2], pad=[3, 3]) with pytest.raises(ValueError): # test when ignore_border and pad >= 0 ds_op = GpuPool(ignore_border=False, ndim=2) ds_op(inp, [2, 2], pad=[1, 1])
def test_pool_big_ws(self): gpu_mode = mode_with_gpu.excluding("cudnn") gpu_mode.check_py_code = False shp = (2, 2, 2, 2) inp = aesara.shared(rand(*shp), "a") inp = aet.as_tensor_variable(inp) ds_op = GpuPool(ignore_border=False, mode="average_exc_pad", ndim=2) pad = aet.as_tensor_variable([0, 0]) f = aesara.function( [], ds_op(inp, [5, 5], stride=[1, 1], pad=pad), mode=gpu_mode ) f()
def test_pool_c_interface(self): gpu_mode = mode_with_gpu.excluding("cudnn") gpu_mode.check_py_code = False shp = (2, 2, 2, 2) inp = aesara.shared(rand(*shp), "a") inp = aet.as_tensor_variable(inp) with pytest.raises(ValueError): # test when ignore_border and pad >= 0 ds_op = GpuPool(ignore_border=False, ndim=2) pad = aet.as_tensor_variable([1, 1]) f = aesara.function([], ds_op(inp, [2, 2], pad=pad), mode=gpu_mode) f()