提交 ca19b615 authored 作者: Harm de Vries's avatar Harm de Vries

update prepare_node blas

上级 5212d681
......@@ -1953,7 +1953,7 @@ class GpuConv(GpuOp):
images[2] * images[3] * 2)
return flops
def prepare_node(self, node):
def prepare_node(self, node, storage_map, compute_map):
if node.op.max_threads_dim0 is None:
cuda = theano.sandbox.cuda
device_id = cuda.use.device_number
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论