提交 b73492ff authored 作者: Yann N. Dauphin's avatar Yann N. Dauphin

missing c_set_tensor4d in GpuDnnConvBase

上级 01232098
......@@ -266,6 +266,9 @@ if (%(err)s != CUDNN_STATUS_SUCCESS) {
}
""" % 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):
desc = inputs[2]
out, = outputs
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论