提交 442d7311 authored 作者: Frederic Bastien's avatar Frederic Bastien

added instanciation of some op.

上级 0624a8dc
...@@ -172,6 +172,7 @@ class GpuCrossentropySoftmaxArgmax1HotWithBias (Op): ...@@ -172,6 +172,7 @@ class GpuCrossentropySoftmaxArgmax1HotWithBias (Op):
#return () #return ()
return (2,) return (2,)
gpu_crossentropy_softmax_argmax_1hot_with_bias = GpuCrossentropySoftmaxArgmax1HotWithBias()
class GpuCrossentropySoftmax1HotWithBiasDx (Op): class GpuCrossentropySoftmax1HotWithBiasDx (Op):
nin=3 nin=3
...@@ -288,6 +289,7 @@ class GpuCrossentropySoftmax1HotWithBiasDx (Op): ...@@ -288,6 +289,7 @@ class GpuCrossentropySoftmax1HotWithBiasDx (Op):
} }
""" % locals() """ % locals()
gpu_crossentropy_softmax_1hot_with_bias_dx = GpuCrossentropySoftmax1HotWithBiasDx()
class GpuSoftmax (Op): class GpuSoftmax (Op):
"""Writeme""" """Writeme"""
...@@ -372,6 +374,7 @@ class GpuSoftmax (Op): ...@@ -372,6 +374,7 @@ class GpuSoftmax (Op):
"}", "}",
]) ])
gpu_softmax = GpuSoftmax()
class GpuSoftmaxWithBias (Op): class GpuSoftmaxWithBias (Op):
"""Writeme""" """Writeme"""
...@@ -474,4 +477,5 @@ class GpuSoftmaxWithBias (Op): ...@@ -474,4 +477,5 @@ class GpuSoftmaxWithBias (Op):
"__syncthreads()", "__syncthreads()",
"}", "}",
]) ])
#for (int i = blockIdx.x; i < N; i += gridDim.x)
gpu_softmax_with_bias = GpuSoftmaxWithBias()
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论