提交 db34c05e authored 作者: abalkin's avatar abalkin

Use NotConstantError instead of TypeError.

上级 6580711f
......@@ -589,7 +589,7 @@ def get_constant_value(v):
v.owner.op.idx_list[0]]:
return numpy.asarray(1)
raise TypeError(v)
raise NotConstantError(v)
class TensorType(Type):
......
......@@ -263,7 +263,7 @@ class RepeatOp(theano.Op):
else:
try:
const_reps = basic.get_constant_value(repeats)
except TypeError:
except basic.NotConstantError:
const_reps = None
if const_reps == 1:
broadcastable = x.broadcastable
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论