提交 a70a1e51 authored 作者: Reyhane Askari's avatar Reyhane Askari

added back outputguard due to pickles

上级 7cf5b30d
...@@ -397,7 +397,7 @@ def get_scalar_constant_value(orig_v, elemwise=True, ...@@ -397,7 +397,7 @@ def get_scalar_constant_value(orig_v, elemwise=True,
"""Return the constant scalar(0-D) value underlying variable `v`. """Return the constant scalar(0-D) value underlying variable `v`.
If `v` is the output of dimshuffles, fills, allocs, rebroadcasts, If `v` is the output of dimshuffles, fills, allocs, rebroadcasts,
cast, DeepCopyOp, ScalarFromTensor, ScalarOp, Elemwise cast, OutputGuard, DeepCopyOp, ScalarFromTensor, ScalarOp, Elemwise
and some pattern with Subtensor, this function digs through them. and some pattern with Subtensor, this function digs through them.
If `v` is not some view of constant scalar data, then raise a If `v` is not some view of constant scalar data, then raise a
...@@ -447,6 +447,9 @@ def get_scalar_constant_value(orig_v, elemwise=True, ...@@ -447,6 +447,9 @@ def get_scalar_constant_value(orig_v, elemwise=True,
max_recur > 0): max_recur > 0):
max_recur -= 1 max_recur -= 1
if isinstance(v.owner.op, (Alloc, DimShuffle, Rebroadcast, if isinstance(v.owner.op, (Alloc, DimShuffle, Rebroadcast,
# outputguard is only used in debugmode but we
# keep it here to avoid problems with old pickels.
compile.ops.OutputGuard,
compile.DeepCopyOp)): compile.DeepCopyOp)):
v = v.owner.inputs[0] v = v.owner.inputs[0]
continue continue
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论