提交 09a441e9 authored 作者: Frederic's avatar Frederic

small fix.

上级 886dc3ad
...@@ -364,7 +364,7 @@ class GpuCrossentropySoftmax1HotWithBiasDx(GpuOp): ...@@ -364,7 +364,7 @@ class GpuCrossentropySoftmax1HotWithBiasDx(GpuOp):
gpu_crossentropy_softmax_1hot_with_bias_dx = GpuCrossentropySoftmax1HotWithBiasDx() gpu_crossentropy_softmax_1hot_with_bias_dx = GpuCrossentropySoftmax1HotWithBiasDx()
class GpuSoftmax(Op): class GpuSoftmax(GpuOp):
""" """
Implement Softmax on the gpu. Implement Softmax on the gpu.
""" """
......
...@@ -255,8 +255,6 @@ def test_sqr_sum_ax0(): ...@@ -255,8 +255,6 @@ def test_sqr_sum_ax0():
z = (x**2).sum(axis=0) z = (x**2).sum(axis=0)
f = theano.function([x], z, mode=mode_without_gpu) f = theano.function([x], z, mode=mode_without_gpu)
f_gpu = theano.function([x], z, mode=mode_with_gpu) f_gpu = theano.function([x], z, mode=mode_with_gpu)
theano.printing.debugprint(f_gpu)
theano.printing.debugprint(f_gpu2)
assert isinstance(f_gpu.maker.fgraph.toposort()[-2].op, assert isinstance(f_gpu.maker.fgraph.toposort()[-2].op,
cuda.nnet.GpuSqrSumAx0) cuda.nnet.GpuSqrSumAx0)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论