提交 600f4833 authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #2241 from ynd/dnn_conv_fix

missing c_set_tensor4d in GpuDnnConvBase
...@@ -266,6 +266,9 @@ if (%(err)s != CUDNN_STATUS_SUCCESS) { ...@@ -266,6 +266,9 @@ if (%(err)s != CUDNN_STATUS_SUCCESS) {
} }
""" % dict(var=var, desc=desc, err=err, fail=fail) """ % dict(var=var, desc=desc, err=err, fail=fail)
def c_set_tensor4d(self, *arg):
return c_set_tensor4d(*arg)
def c_code(self, node, name, inputs, outputs, sub): def c_code(self, node, name, inputs, outputs, sub):
desc = inputs[2] desc = inputs[2]
out, = outputs out, = outputs
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论