提交 9bb0691f authored 作者: Frederic Bastien's avatar Frederic Bastien

optimization local_subtensor_make_vector don't use the shape_feature.

上级 b3fef2e9
......@@ -930,9 +930,6 @@ def local_subtensor_make_vector(node):
# we can do this for constant indexes
if isinstance(node.op, T.Subtensor):
# This optimization needs ShapeOpt and env.shape_feature
if not hasattr(node.env, 'shape_feature'):
return
shape_feature = node.env.shape_feature
x = node.inputs[0]
if x.owner and x.owner.op == make_vector:
try:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论