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

Use NotConstantError instead of TypeError.

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