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

small fix.

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