提交 6c4c5408 authored 作者: David Warde-Farley's avatar David Warde-Farley

PEP 3113 cleanup of GpuConv*3D.py

上级 fd8541cb
...@@ -38,7 +38,8 @@ class GpuConv3D(theano.Op): ...@@ -38,7 +38,8 @@ class GpuConv3D(theano.Op):
def c_code_cache_version(self): def c_code_cache_version(self):
return () return ()
def c_code(self, node, nodename, (V,W,b,d), outputs, sub): def c_code(self, node, nodename, inputs, outputs, sub):
V, W, b, d = inputs
fail = sub['fail'] fail = sub['fail']
H = outputs[0] H = outputs[0]
......
...@@ -66,7 +66,8 @@ class GpuConvGrad3D(theano.Op): ...@@ -66,7 +66,8 @@ class GpuConvGrad3D(theano.Op):
output_storage[0][0] = dCdW output_storage[0][0] = dCdW
def c_code(self, node, nodename, (V,d,WShape,dCdH), outputs, sub): def c_code(self, node, nodename, inputs, outputs, sub):
V, d, WShape, dCdH = inputs
fail = sub['fail'] fail = sub['fail']
dCdW = outputs[0] dCdW = outputs[0]
......
...@@ -48,7 +48,8 @@ class GpuConvTransp3D(theano.Op): ...@@ -48,7 +48,8 @@ class GpuConvTransp3D(theano.Op):
def c_code_cache_version(self): def c_code_cache_version(self):
return () return ()
def c_code(self, node, nodename, (W, b, d, H, RShape), outputs, sub): def c_code(self, node, nodename, inputs, outputs, sub):
W, b, d, H, RShape = inputs
fail = sub['fail'] fail = sub['fail']
R = outputs[0] R = outputs[0]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论