提交 f32dff50 authored 作者: Pierre Luc Carrier's avatar Pierre Luc Carrier

Modify parent class of GpuImages2Neibs

上级 89178cd5
# This is work in progress
from theano import Op, Apply
from theano.gof import local_optimizer
from theano.sandbox.cuda import cuda_available, GpuOp
from theano.sandbox.cuda import cuda_available
from theano.sandbox.neighbours import Images2Neibs
......@@ -11,7 +11,7 @@ if cuda_available:
from theano.sandbox.cuda.opt import register_opt as register_gpu_opt
class GpuImages2Neibs(Images2Neibs, GpuOp):
class GpuImages2Neibs(Images2Neibs, Op):
def __init__(self, mode='valid'):
if mode not in ['valid', 'ignore_borders', 'wrap_centered']:
raise NotImplementedError("Only the mode valid, ignore_borders"
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论