提交 117558b5 authored 作者: carriepl's avatar carriepl

For test_infer_shape2 to use FAST_RUN mode

上级 7e777e65
...@@ -2580,7 +2580,8 @@ class T_Scan(unittest.TestCase): ...@@ -2580,7 +2580,8 @@ class T_Scan(unittest.TestCase):
None], None],
n_steps=5) n_steps=5)
f = theano.function([seq, sitsot_init, mitsot_init], out[2].shape) f = theano.function([seq, sitsot_init, mitsot_init], out[2].shape,
mode='FAST_RUN')
assert(len(scan_nodes_from_fct(f)) == 0) assert(len(scan_nodes_from_fct(f)) == 0)
output_shape = f(numpy.arange(5), 5, [1, 2]) output_shape = f(numpy.arange(5), 5, [1, 2])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论