提交 79597a04 authored 作者: carriepl's avatar carriepl

Ensure that test depending on shape optimizations uses mode including them

上级 b57c6478
...@@ -3977,7 +3977,8 @@ class T_Scan(unittest.TestCase): ...@@ -3977,7 +3977,8 @@ class T_Scan(unittest.TestCase):
# This fails if Scan's infer_shape() is unable to remove the Scan # This fails if Scan's infer_shape() is unable to remove the Scan
# node from the graph. # node from the graph.
f_infershape = theano.function([seq], results[1].shape) f_infershape = theano.function([seq], results[1].shape,
mode='FAST_RUN')
scan_nodes_infershape = scan_nodes_from_fct(f_infershape) scan_nodes_infershape = scan_nodes_from_fct(f_infershape)
assert(len(scan_nodes_infershape) == 0) assert(len(scan_nodes_infershape) == 0)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论