提交 ae8ce3c6 authored 作者: Pierre Luc Carrier's avatar Pierre Luc Carrier

Correct various bugs with ops GpuSoftmax and GpuSoftmaxWithBias. Current version…

Correct various bugs with ops GpuSoftmax and GpuSoftmaxWithBias. Current version passes tests with float32
上级 6ef5d264
差异被折叠。
......@@ -344,12 +344,12 @@ def local_gpua_crossentropysoftmax1hotwithbiasdx(node):
return GpuCrossentropySoftmax1HotWithBiasDx()
@register_opt()
@op_lifter([tensor.nnet.GpuSoftmax])
@op_lifter([tensor.nnet.Softmax])
def local_gpua_softmax(node):
return GpuSoftmax()
@register_opt()
@op_lifter([tensor.nnet.GpuSoftmaxWithBias])
@op_lifter([tensor.nnet.SoftmaxWithBias])
def local_gpua_softmaxwithbias(node):
return GpuSoftmaxWithBias()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论