提交 eb1d9063 authored 作者: Pierre Luc Carrier's avatar Pierre Luc Carrier

Make conditions more precise

上级 63a7a5dc
...@@ -729,7 +729,7 @@ class PushOutScanOutput(gof.Optimizer): ...@@ -729,7 +729,7 @@ class PushOutScanOutput(gof.Optimizer):
# 1 if sitsot_in_idx==0 # 1 if sitsot_in_idx==0
dot_input = nd.inputs[dot_in_idx] dot_input = nd.inputs[dot_in_idx]
if (isinstance(dot_input.owner, gof.Apply) and if (dot_input.owner is not None and
isinstance(dot_input.owner.op, theano.tensor.Dot) and isinstance(dot_input.owner.op, theano.tensor.Dot) and
len(dot_input.clients) == 1 and len(dot_input.clients) == 1 and
dot_input.owner.inputs[0].ndim == 2 and dot_input.owner.inputs[0].ndim == 2 and
...@@ -797,7 +797,7 @@ class PushOutScanOutput(gof.Optimizer): ...@@ -797,7 +797,7 @@ class PushOutScanOutput(gof.Optimizer):
client = outer_var.clients[0][0] client = outer_var.clients[0][0]
if (isinstance(client, gof.Apply) and if (client != 'output' and
isinstance(client.op, theano.tensor.Subtensor) and isinstance(client.op, theano.tensor.Subtensor) and
isinstance(client.inputs[1], theano.Constant) and isinstance(client.inputs[1], theano.Constant) and
client.inputs[1].ndim == 0 and client.inputs[1].ndim == 0 and
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论