提交 70aac84f authored 作者: Ozan Çağlayan's avatar Ozan Çağlayan

sandbox/cuda: add old bad names to the end

上级 bda9859c
...@@ -1352,7 +1352,8 @@ def cast(x, dtype): ...@@ -1352,7 +1352,8 @@ def cast(x, dtype):
@register_opt() @register_opt()
@local_optimizer([tensor.nnet.CrossentropySoftmaxArgmax1HotWithBias]) @local_optimizer([tensor.nnet.CrossentropySoftmaxArgmax1HotWithBias],
'local_gpu_crossentorpy_softmax_argmax_1hot_with_bias')
def local_gpu_crossentropy_softmax_argmax_1hot_with_bias(node): def local_gpu_crossentropy_softmax_argmax_1hot_with_bias(node):
if isinstance(node.op, tensor.nnet.CrossentropySoftmaxArgmax1HotWithBias): if isinstance(node.op, tensor.nnet.CrossentropySoftmaxArgmax1HotWithBias):
x, b, y = node.inputs x, b, y = node.inputs
...@@ -1381,7 +1382,8 @@ def local_gpu_crossentropy_softmax_argmax_1hot_with_bias(node): ...@@ -1381,7 +1382,8 @@ def local_gpu_crossentropy_softmax_argmax_1hot_with_bias(node):
@register_opt() @register_opt()
@local_optimizer([tensor.nnet.CrossentropySoftmax1HotWithBiasDx]) @local_optimizer([tensor.nnet.CrossentropySoftmax1HotWithBiasDx],
'local_gpu_crossentorpy_softmax_1hot_with_bias_dx')
def local_gpu_crossentropy_softmax_1hot_with_bias_dx(node): def local_gpu_crossentropy_softmax_1hot_with_bias_dx(node):
if isinstance(node.op, tensor.nnet.CrossentropySoftmax1HotWithBiasDx): if isinstance(node.op, tensor.nnet.CrossentropySoftmax1HotWithBiasDx):
dnll, sm, yidx = node.inputs dnll, sm, yidx = node.inputs
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论