Skip to content

P
pytensor
提交 b47b95c3 authored 作者: Pierre Luc Carrier's avatar Pierre Luc Carrier
浏览文件
操作

Convert CudaNdarray_DEV_DATA to PyGpuArray_DATA in ops GpuSoftmax and GpuSoftmaxWithBias.

上级 1c0aff18
显示空白字符变更
内嵌 并排
正在显示 包含 10 行增加 和 10 行删除
theano/sandbox/gpuarray/nnet.py
浏览文件 @ b47b95c3
...@@ -521,11 +521,11 @@ class GpuSoftmax (Op): ...@@ -521,11 +521,11 @@ class GpuSoftmax (Op):
PyGpuArray_DIMS(%(x)s)[0], PyGpuArray_DIMS(%(x)s)[0],
PyGpuArray_DIMS(%(x)s)[1], PyGpuArray_DIMS(%(x)s)[1],
CudaNdarray_DEV_DATA(%(x)s), PyArray_DATA(%(x)s),
PyGpuArray_STRIDES(%(x)s)[0], PyGpuArray_STRIDES(%(x)s)[0],
PyGpuArray_STRIDES(%(x)s)[1], PyGpuArray_STRIDES(%(x)s)[1],
CudaNdarray_DEV_DATA(%(z)s), PyArray_DATA(%(z)s),
PyGpuArray_STRIDES(%(z)s)[0], PyGpuArray_STRIDES(%(z)s)[0],
PyGpuArray_STRIDES(%(z)s)[1] PyGpuArray_STRIDES(%(z)s)[1]
); );
...@@ -539,11 +539,11 @@ class GpuSoftmax (Op): ...@@ -539,11 +539,11 @@ class GpuSoftmax (Op):
PyGpuArray_DIMS(%(x)s)[0], PyGpuArray_DIMS(%(x)s)[0],
PyGpuArray_DIMS(%(x)s)[1], PyGpuArray_DIMS(%(x)s)[1],
CudaNdarray_DEV_DATA(%(x)s), PyArray_DATA(%(x)s),
PyGpuArray_STRIDES(%(x)s)[0], PyGpuArray_STRIDES(%(x)s)[0],
PyGpuArray_STRIDES(%(x)s)[1], PyGpuArray_STRIDES(%(x)s)[1],
CudaNdarray_DEV_DATA(%(z)s), PyArray_DATA(%(z)s),
PyGpuArray_STRIDES(%(z)s)[0], PyGpuArray_STRIDES(%(z)s)[0],
PyGpuArray_STRIDES(%(z)s)[1] PyGpuArray_STRIDES(%(z)s)[1]
); );
...@@ -703,14 +703,14 @@ class GpuSoftmaxWithBias (GpuOp): ...@@ -703,14 +703,14 @@ class GpuSoftmaxWithBias (GpuOp):
PyGpuArray_DIMS(%(x)s)[0], PyGpuArray_DIMS(%(x)s)[0],
PyGpuArray_DIMS(%(x)s)[1], PyGpuArray_DIMS(%(x)s)[1],
CudaNdarray_DEV_DATA(%(x)s), PyArray_DATA(%(x)s),
PyGpuArray_STRIDES(%(x)s)[0], PyGpuArray_STRIDES(%(x)s)[0],
PyGpuArray_STRIDES(%(x)s)[1], PyGpuArray_STRIDES(%(x)s)[1],
CudaNdarray_DEV_DATA(%(b)s), PyArray_DATA(%(b)s),
PyGpuArray_STRIDES(%(b)s)[0], PyGpuArray_STRIDES(%(b)s)[0],
CudaNdarray_DEV_DATA(%(z)s), PyArray_DATA(%(z)s),
PyGpuArray_STRIDES(%(z)s)[0], PyGpuArray_STRIDES(%(z)s)[0],
PyGpuArray_STRIDES(%(z)s)[1] PyGpuArray_STRIDES(%(z)s)[1]
); );
...@@ -724,14 +724,14 @@ class GpuSoftmaxWithBias (GpuOp): ...@@ -724,14 +724,14 @@ class GpuSoftmaxWithBias (GpuOp):
PyGpuArray_DIMS(%(x)s)[0], PyGpuArray_DIMS(%(x)s)[0],
PyGpuArray_DIMS(%(x)s)[1], PyGpuArray_DIMS(%(x)s)[1],
CudaNdarray_DEV_DATA(%(x)s), PyArray_DATA(%(x)s),
PyGpuArray_STRIDES(%(x)s)[0], PyGpuArray_STRIDES(%(x)s)[0],
PyGpuArray_STRIDES(%(x)s)[1], PyGpuArray_STRIDES(%(x)s)[1],
CudaNdarray_DEV_DATA(%(b)s), PyArray_DATA(%(b)s),
PyGpuArray_STRIDES(%(b)s)[0], PyGpuArray_STRIDES(%(b)s)[0],
CudaNdarray_DEV_DATA(%(z)s), PyArray_DATA(%(z)s),
PyGpuArray_STRIDES(%(z)s)[0], PyGpuArray_STRIDES(%(z)s)[0],
PyGpuArray_STRIDES(%(z)s)[1] PyGpuArray_STRIDES(%(z)s)[1]
); );
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论