提交 25dcaef4 authored 作者: Nan Rosemary Ke's avatar Nan Rosemary Ke

remove name argument also from GpuReshape

上级 911fa71a
...@@ -643,7 +643,7 @@ def local_gpua_contiguous(op, context_name, inputs, outputs): ...@@ -643,7 +643,7 @@ def local_gpua_contiguous(op, context_name, inputs, outputs):
@op_lifter([tensor.Reshape]) @op_lifter([tensor.Reshape])
@register_opt2([tensor.Reshape], 'fast_compile') @register_opt2([tensor.Reshape], 'fast_compile')
def local_gpua_reshape(op, context_name, inputs, outputs): def local_gpua_reshape(op, context_name, inputs, outputs):
res = GpuReshape(op.ndim, None) res = GpuReshape(op.ndim)
return res return res
...@@ -662,7 +662,7 @@ def local_gpua_flatten(op, context_name, inputs, outputs): ...@@ -662,7 +662,7 @@ def local_gpua_flatten(op, context_name, inputs, outputs):
if op.outdim != 1: if op.outdim != 1:
shp = [inputs[0].shape[i] for i in range(op.outdim - 1)] shp = [inputs[0].shape[i] for i in range(op.outdim - 1)]
shp += [-1] shp += [-1]
res = GpuReshape(op.outdim, None) res = GpuReshape(op.outdim)
o = res(inputs[0], theano.tensor.as_tensor_variable(shp)) o = res(inputs[0], theano.tensor.as_tensor_variable(shp))
return o return o
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论