提交 251f616f authored 作者: Frederic Bastien's avatar Frederic Bastien

Update conv doc.

上级 d844e6c1
...@@ -2134,7 +2134,7 @@ def local_gpua_abstractconv(op, context_name, inputs, outputs): ...@@ -2134,7 +2134,7 @@ def local_gpua_abstractconv(op, context_name, inputs, outputs):
AbstractConv2d_gradInputs, AbstractConv2d_gradInputs,
AbstractConv3d, AbstractConv3d,
AbstractConv3d_gradWeights, AbstractConv3d_gradWeights,
AbstractConv3d_gradInputs], 'fast_compile', 'conv_dnn', 'cudnn') AbstractConv3d_gradInputs], 'fast_compile')
def local_gpua_lift_abstractconv_graph(op, context_name, inputs, outputs): def local_gpua_lift_abstractconv_graph(op, context_name, inputs, outputs):
inps = list(inputs) inps = list(inputs)
inps[0] = as_gpuarray_variable(inputs[0], inps[0] = as_gpuarray_variable(inputs[0],
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论