提交 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):
def c_code_cache_version(self):
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']
H = outputs[0]
......
......@@ -66,7 +66,8 @@ class GpuConvGrad3D(theano.Op):
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']
dCdW = outputs[0]
......
......@@ -48,7 +48,8 @@ class GpuConvTransp3D(theano.Op):
def c_code_cache_version(self):
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']
R = outputs[0]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论