Skip to content

P
pytensor
提交 fec5c70e authored 作者: Harm de Vries's avatar Harm de Vries
浏览文件
操作

fix

上级 e1c65a56
显示空白字符变更
内嵌 并排
正在显示 包含 6 行增加 和 6 行删除
theano/sandbox/cuda/dnn.py
浏览文件 @ fec5c70e
...@@ -1390,19 +1390,19 @@ class GpuDnnPool(DnnBase): ...@@ -1390,19 +1390,19 @@ class GpuDnnPool(DnnBase):
node.inputs[1] = ws node.inputs[1] = ws
node.inputs.append(st) node.inputs.append(st)
node.inputs.append(pad) node.inputs.append(pad)
if isinstance(ws, theano.constant): if isinstance(ws, theano.Constant):
storage_map[ws] = [ws.data] storage_map[ws] = [ws.data]
compute_map[ws] = [True] compute_map[ws] = [True]
else: else:
storage_map[ws] = [None] storage_map[ws] = [None]
compute_map[ws] = [False] compute_map[ws] = [False]
if isinstance(st, theano.constant): if isinstance(st, theano.Constant):
storage_map[st] = [st.data] storage_map[st] = [st.data]
compute_map[st] = [True] compute_map[st] = [True]
else: else:
storage_map[st] = [None] storage_map[st] = [None]
compute_map[st] = [False] compute_map[st] = [False]
if isinstance(pad, theano.constant): if isinstance(pad, theano.Constant):
storage_map[pad] = [pad.data] storage_map[pad] = [pad.data]
compute_map[pad] = [True] compute_map[pad] = [True]
else: else:
...@@ -1627,19 +1627,19 @@ class GpuDnnPoolGrad(DnnBase): ...@@ -1627,19 +1627,19 @@ class GpuDnnPoolGrad(DnnBase):
node.inputs[3] = ws node.inputs[3] = ws
node.inputs.append(st) node.inputs.append(st)
node.inputs.append(pad) node.inputs.append(pad)
if isinstance(ws, theano.constant): if isinstance(ws, theano.Constant):
storage_map[ws] = [ws.data] storage_map[ws] = [ws.data]
compute_map[ws] = [True] compute_map[ws] = [True]
else: else:
storage_map[ws] = [None] storage_map[ws] = [None]
compute_map[ws] = [False] compute_map[ws] = [False]
if isinstance(st, theano.constant): if isinstance(st, theano.Constant):
storage_map[st] = [st.data] storage_map[st] = [st.data]
compute_map[st] = [True] compute_map[st] = [True]
else: else:
storage_map[st] = [None] storage_map[st] = [None]
compute_map[st] = [False] compute_map[st] = [False]
if isinstance(pad, theano.constant): if isinstance(pad, theano.Constant):
storage_map[pad] = [pad.data] storage_map[pad] = [pad.data]
compute_map[pad] = [True] compute_map[pad] = [True]
else: else:
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论