Skip to content

P
pytensor
提交 0b2f4053 authored 作者: xiaoqie's avatar xiaoqie
浏览文件
操作

fix type

上级 ea0aff00
显示空白字符变更
内嵌 并排
正在显示 包含 2 行增加 和 2 行删除
theano/gpuarray/nnet.py
浏览文件 @ 0b2f4053
...@@ -597,7 +597,7 @@ class GpuSoftmax(GpuKernelBase, Op): ...@@ -597,7 +597,7 @@ class GpuSoftmax(GpuKernelBase, Op):
type_sm = gpuarray.dtype_to_ctype(dtype_sm) type_sm = gpuarray.dtype_to_ctype(dtype_sm)
type_acc = gpuarray.dtype_to_ctype(work_sm) type_acc = gpuarray.dtype_to_ctype(work_sm)
ctype = gpuarray.dtype_to_ctype(dtype_sm) ctype = gpuarray.dtype_to_ctype(work_sm)
params = [ params = [
gpuarray.SIZE, gpuarray.SIZE, gpuarray.SIZE, gpuarray.SIZE,
...@@ -915,7 +915,7 @@ class GpuSoftmaxWithBias(GpuKernelBase, Op): ...@@ -915,7 +915,7 @@ class GpuSoftmaxWithBias(GpuKernelBase, Op):
type_sm = gpuarray.dtype_to_ctype(dtype_sm) type_sm = gpuarray.dtype_to_ctype(dtype_sm)
type_acc = gpuarray.dtype_to_ctype(work_sm) type_acc = gpuarray.dtype_to_ctype(work_sm)
ctype = gpuarray.dtype_to_ctype(dtype_sm) ctype = gpuarray.dtype_to_ctype(work_sm)
params = [ params = [
gpuarray.SIZE, gpuarray.SIZE, gpuarray.SIZE, gpuarray.SIZE,
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论