提交 1bc89154 authored 作者: Frederic Bastien's avatar Frederic Bastien 提交者: sentient07

Use the good name, but it isn't used, just clarification.

上级 78aa9113
......@@ -757,7 +757,7 @@ def local_gpua_specifyShape(op, context_name, inputs, outputs):
@register_opt('fast_compile')
@op_lifter([theano.compile.ops.Shape])
@register_opt2([tensor.compile.ops.Shape], 'fast_compile')
def local_gpua_shape(node, context_name, inputs, outputs):
def local_gpua_shape(op, context_name, inputs, outputs):
# op_lifter will call this opt too frequently as the output is
# always on the CPU.
if isinstance(inputs[0].type, GpuArrayType):
......@@ -1215,7 +1215,7 @@ def local_gpua_softmax(op, context_name, inputs, outputs):
@register_opt('fast_compile')
@op_lifter([tensor.nnet.SoftmaxWithBias], cuda_only=True)
@register_opt2([tensor.nnet.SoftmaxWithBias], 'fast_compile')
def local_gpua_softmaxwithbias(node, context_name, inputs, outputs):
def local_gpua_softmaxwithbias(op, context_name, inputs, outputs):
return gpu_softmax_with_bias
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论